Java 将文件绑定到标准输入

发表于 更新于 分类于 阅读次数：

有些算法题用这种方式更容易测试。

有些算法题需要很复杂形式的系统输入，如果采用将文件绑定到标准输入的方式，可以更方便地测试代码。

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Scanner;

class Main {
    public static void main(String[] args) {
        File file = new File("test_case_1.txt");
        try {
            FileInputStream fileInputStream = new FileInputStream(file);
            System.setIn(fileInputStream);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        
        //...
    }
}

测试多组测试用例

使用 while (sc.hasNextInt())

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package test2;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Scanner;

class Main {
    public static void main(String[] args) {
        File file = new File("test_case_1.txt");
        try {
            FileInputStream fileInputStream = new FileInputStream(file);
            System.setIn(fileInputStream);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }

        Scanner sc = new Scanner(System.in);
        while (sc.hasNextInt()) {
            //...
        }
    }
}