LeetCode 3153 - 所有数对中数位差之和

class Solution {
public:
    using ll = long long;
    long long sumDigitDifferences(vector<int>& nums) {
        // digits[i][j]表示左侧所有数字中，第i位数字为j的个数
        vector<vector<ll>> digits(10, vector<ll>(10));
        ll ans = 0;
        for (int i = 0; i < nums.size(); ++i) {
            int num = nums[i];
            int idx = 0;
            while (num) {
                int digit = num % 10;

                for (int k = 0; k < 10; ++k) {
                    if (k == digit) continue;  // 与当前数字该位的数字相同的不算
                    ans += digits[idx][k];  // 累加左侧所有数字中，第idx位不同于当前数字该位的数字出现次数
                }

                ++digits[idx][digit];  // 更新digits数组
                ++idx; num /= 10;
            }
        }
        return ans;
    }
};