1814. Count Nice Pairs in an Array

• 원래 숫자와 뒤집은 숫자의 합이 같은 두 숫자의 쌍 개수를 구하는 문제.
• 조건을 표현하면 아래와 같다.
• nums[i] + rev(nums[j]) == rev(nums[i]) + nums[j]
• 위 조건을 변형하면 아래와 같다.
• nums[i] - rev(nums[i]) == nums[j] - rev(nums[j])
• 한마디로 말하면 원래 숫자, 뒤집은 숫자의 차이가 같은 두 index 쌍의 개수를 구하면 된다.

• 풀이는 다음과 같이 이루어진다.
• 먼저 원래 숫자, 뒤집은 숫자의 차이를 계산한다. 원래 숫자는 필요없기 때문에 nums에 차이 값을 다시 저장한다.
• 그 차이 값 숫자를 정렬한다.
• 마지막에 더미 값 (마지막 값 + 1)을 추가한다.
• 루프 밖에서 마지막 값 항목에 대한 값을 따로 계산해도 되지만 중복 코드가 생기므로…
• nums에 있는 차이 값들을 순회하면서 동일한 값들의 숫자를 세고 이를 바탕으로 index 쌍 조합의 개수를 계산한다.
• nums가 정렬되어있기 때문에 동일한 값들 끼리 붙어있게 된다. 따라서 값이 달라질 때까지 숫자의 개수를 세면 동일한 값들의 숫자를 셀 수 있다.
• 차이 값이 동일한 k개의 index가 있을때 그 중에 2개 값을 골라서 쌍을 만들면 k x (k - 1) / 2 개가 된다.
• 곱하기 연산에서 overflow가 발생할 수 있으므로 long long 으로 casting 한다.
• 곱한 결과, 그리고 결과 값 더하는 부분에서 overflow가 발생할 수 있으므로 1e9 + 7로 모듈러 연산을 한다.

class Solution {
public:
    static constexpr int MOD = 1e9 + 7;

    int rev(int v) {
        int result = 0;
        while (v > 0) {
            result *= 10;
            result += v % 10;
            v /= 10;
        }
        return result;
    }

    int countNicePairs(vector<int>& nums) {
        for (int i = 0; i < nums.size(); ++i) {
            nums[i] = nums[i] - rev(nums[i]);
        }
        sort(nums.begin(), nums.end());
        nums.push_back(nums.back() + 1);
        int count = 0;
        int prev = 0;
        int result = 0;
        for (auto diff : nums) {
            if (diff != prev) {
                int subResult = (
                    count * static_cast<long long>(count - 1) / 2
                ) % MOD;
                result = (result + subResult) % MOD;
                count = 0;
            }
            prev = diff;
            ++count;
        }
        return result;
    }
};