// 提出時にassertはオフ #ifndef DEBUG #ifndef NDEBUG #define NDEBUG #endif #endif #include using namespace std; using ll = long long; #define ALL(x) (x).begin(), (x).end() template using vec = vector; int N, P; ll ans = 0; // 同じmodPの集合に対し、そのmodを引いて、Pで割る void groupByModP(const vec &A) { if(A.size() <= 1) return; int curMod = A[0] % P; unordered_map> newGroup; for(int num : A) { int newNum = num + P - curMod; assert(newNum % P == 0); newNum /= P; newGroup[newNum % P].push_back(newNum); } ans += (ll)A.size() * (A.size() - 1) / 2; for(const pair> &p : newGroup) { groupByModP(p.second); } } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cin >> N >> P; vec A(N); for(int &a : A) cin >> a; unordered_map> group; for(int num : A) { group[num % P].push_back(num); } for(const pair> &p : group) { groupByModP(p.second); } cout << ans << "\n"; }