小猴编程周赛C++ | 2023年春第10周T2 K的倍数（二）

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】

给定包含 n 个正整数的数列 a1,a2,…,an，解答如下问题：

对于所有满足 1≤i<j≤n 的整数组 (i,j)，求出所有满足"ai+aj 是 k 的倍数"的整数组 (i,j) 的个数。

【输入】

第一行，包含两个正整数 n,k。

第二行，包含 n 个正整数 a1,a2...,an。

【输出】

一行，一个整数表示结果。

【输入样例】

8 2
1 2 3 4 5 6 7 2

【输出样例】

12

【代码详解】

#include <bits/stdc++.h>
using namespace std;
int n, k;
int a[200005]={0};
long long cnt[200005]={0};
long long ans=0;
int main()
{
    cin >> n >> k;
    for (int i=1; i<=n; i++) {
        cin >> a[i];
        cnt[a[i]%k]++;
    }
    /*
    k=6
    0 1 2 3 ... k-1
    0和3要和自己凑，其他，如1与k-1，2与k-2凑...
    */
    for (int i=0; i<=k/2; i++) {
        if ((i+i)%k==0) ans += cnt[i]*(cnt[i]-1)/2;
        else ans += cnt[i]*cnt[k-i];
    }
    cout << ans;
    return 0;
}

【运行结果】

8 2
1 2 3 4 5 6 7 2
12