792. 高精度减法Subtracts two large numbers represented as reverse

792. 高精度减法 - AcWing题库

Question

Content

给定两个正整数（不含前导 0），计算它们的差，计算结果可能为负数。

输入格式

共两行，每行包含一个整数。

输出格式

共一行，包含所求的差。

数据范围

1≤整数长度≤ $10^5$

输入样例：

32
11

输出样例：

Solution

Java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));

        String a = bufferedReader.readLine();
        List<Integer> list = new ArrayList<>();
        for (int i = a.length() - 1; i >= 0; i--) {
            list.add(a.charAt(i) - '0');
        }

        String b = bufferedReader.readLine();
        List<Integer> list2 = new ArrayList<>();
        for (int i = b.length() - 1; i >= 0; i--) {
            list2.add(b.charAt(i) - '0');
        }

        List<Integer> ans = subtract(list, list2);

        for (int i = ans.size() - 1; i >= 0; i--) {
            System.out.print(ans.get(i));
        }

        bufferedReader.close();
    }

    private static List<Integer> subtract(List<Integer> a, List<Integer> b) {
        if (!compare(a, b)) {
            System.out.print("-");
            return subtract(b, a);
        }

        int t = 0;

        List<Integer> c = new ArrayList<>();

        for (int i = 0; i < a.size(); i++) {
            t = a.get(i) - t;

            if (i < b.size()) {
                t -= b.get(i);
            }

            c.add((t + 10) % 10);

            if (t < 0) {
                t = 1;
            } else {
                t = 0;
            }
        }

        while (c.size() > 1 && c.get(c.size() - 1) == 0) {
            c.remove(c.size() - 1);
        }

        return c;
    }


    private static boolean compare(List<Integer> a, List<Integer> b) {
        if (a.size() != b.size()) {
            return a.size() > b.size();
        }

        for (int i = a.size() - 1; i >= 0; i--) {
            if (a.get(i).equals(b.get(i))) {
                continue;
            }

            return a.get(i) > b.get(i);
        }

        return true;
    }

}