对拍程序-java

import util.RandomArrList;

import java.io.*;
import java.lang.reflect.Method;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

public class CompareCode {

    private static final String TEST_CASES_FOLDER = "./"; // 当前目录

    public static void main(String[] args) throws IOException {
        int numCase = 1;

        // 生成测试用例数据并写入文件
        generateTestCases(numCase);

        // 调用算法 A 和算法 B 进行对拍
        for (int i = 1; i <= numCase; i++) {
            String inputFilePath = TEST_CASES_FOLDER + "testcase" + i + ".txt";
            String outputAFilePath = TEST_CASES_FOLDER + "outputA" + i + ".txt";
            String outputBFilePath = TEST_CASES_FOLDER + "outputB" + i + ".txt";

            runAlgorithm(A.class, inputFilePath, outputAFilePath);
            runAlgorithm(B.class, inputFilePath, outputBFilePath);

            if (!compareOutputs(outputAFilePath, outputBFilePath)) {
                System.setOut(System.err);
                System.out.println("Output mismatch for testcase" + i);
                System.out.println("A: " + outputAFilePath);
                System.out.println("B: " + outputBFilePath);
                System.setOut(System.out);
                System.out.println("============================================");
                System.out.println();
            }
        }
    }

    // 调用程序处理输入文件并将输出结果写入文件
    private static void runAlgorithm(Class<?> clazz, String inputFilePath, String outputFilePath) {
        try {
            Method method = clazz.getMethod("main", String[].class);

            // 读取输入文件内容
            List<String> inputFileContent = Files.readAllLines(Paths.get(inputFilePath));

            // 设置 System.in 为读取输入文件内容的流
            ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(String.join(System.lineSeparator(), inputFileContent).getBytes(StandardCharsets.UTF_8));
            System.setIn(byteArrayInputStream);

            // 设置 System.out 为输出到文件
            PrintStream fileOut = new PrintStream(new FileOutputStream(outputFilePath));
            System.setOut(fileOut);

            // 调用算法的 main 方法，并将空的参数数组传递给算法
            method.invoke(null, new Object[]{new String[]{}});

            // 恢复 System.in 和 System.out
            System.setIn(System.in);
            System.setOut(System.out);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    // 比较两个输出文件的内容是否一致
    private static boolean compareOutputs(String outputFilePath1, String outputFilePath2) throws IOException {
        byte[] bytes1 = Files.readAllBytes(Paths.get(outputFilePath1));
        byte[] bytes2 = Files.readAllBytes(Paths.get(outputFilePath2));
        String content1 = new String(bytes1, StandardCharsets.UTF_8);
        String content2 = new String(bytes2, StandardCharsets.UTF_8);

        return content1.equals(content2);
    }


    // 生成测试用例数据并写入文件
    private static void generateTestCases1(int numTestCases) throws IOException {
        for (int i = 1; i <= numTestCases; i++) {
            String filePath = TEST_CASES_FOLDER + "testcase" + i + ".txt";
            try (BufferedWriter writer = new BufferedWriter(new FileWriter(filePath))) {
                int n = RandomArrList.getRandNum(3, 10);
                writer.write(n + "\n");
                int[] randomArr = RandomArrList.getRandomArr(n, 1, n);
                for (int j = 0; j < randomArr.length; j++) {
                    writer.write(randomArr[j] + " ");
                }
                writer.write("\n");
            }
        }
    }

    // 生成测试用例数据并写入文件
    private static void generateTestCases(int numTestCases) throws IOException {

        for (int i = 1; i <= numTestCases; i++) {
            String filePath = TEST_CASES_FOLDER + "testcase" + i + ".txt";
            try (BufferedWriter writer = new BufferedWriter(new FileWriter(filePath))) {
                int t = RandomArrList.getRandNum(1, 10);
                writer.write(t + "\n");
                for (int j = 0; j < t; j++) {
                    int n = RandomArrList.getRandNum(1, 15);
                    int m = RandomArrList.getRandNum(1, 15);
                    writer.write(n + " " + m);
                    writer.write("\n");
                    ArrayList<Integer> randomList = RandomArrList.getRandomList(n, 1, 10);
                    for (Integer num : randomList) {
                        writer.write(num + " ");
                    }
                    writer.write("\n");
                }
            }
        }
    }
}

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Random;

/**
 * 生成随机数组
 *
 **/
public class RandomArrList {
	
    // 生成随机数数组，长度为length，最小数为start，最大数为end
    public static int[] getRandomArr(int length, int start, int end) {
        // T[] ts = (T[])Array.newInstance(componentType, length);
        int[] resArr = new int[length];
        Random random = new Random();
        for (int i = 0; i < length; i++) {
            int num = random.nextInt(end - start + 1) + start;
            resArr[i] = num;
        }
        return resArr;
    }

    // 生成随机数集合，长度为length，最小数为start，最大数为end
    public static ArrayList<Integer> getRandomList(int length, int start, int end) {
        ArrayList<Integer> resList = new ArrayList<>(length);
        Random random = new Random();
        for (int i = 0; i < length; i++) {
            int num = random.nextInt(end - start + 1) + start;
            resList.add(num);
        }
        return resList;
    }
	
    // 生成单个随机数
    public static int getRandNum(int start, int end) {
        Random random = new Random();
        return random.nextInt(end - start + 1) + start;
    }
}