时间限制 1000 ms 内存限制 64 MB
题目描述
给你一个长度为n的数组和一个正整数k,问从数组中任选两个数使其和是k的倍数,有多少种选法 对于数组a1=1 , a2=2 , a3=2而言: (a1,a2)和(a2,a1)被认为是同一种选法; (a1,a2)和(a1,a3)被认为是不同的选法。
输入数据
第一行有两个正整数n,k。n<=1000000,k<=1000000 第二行有n个正整数,每个数的大小不超过1e9
输出数据
选出一对数使其和是k的倍数的选法个数
样例输入
5 6 1 2 3 4 5
样例输出
2
样例说明
样例解释: a1+a5=6,a2+a4=6,都是6的倍数 所以符合条件的选法有(1,5),(2,4)
#include <iostream> #include <algorithm> #include <cstdio> #include <cmath> using namespace std; int n, k; long long ans = 0; int l[1000001] = { 0 }; int main() { //cin >> n >> k; scanf("%d%d", &n, &k); for (int i = 0; i < n; i++) { int tmp; scanf("%d", &tmp); //cin >> tmp; l[tmp % k]++; } ans += l[0] * (l[0] - 1) / 2; //for (int i = 1, j = k - 1; i < j; i++, j--) { // ans += l[i] * l[j]; //} for (int i = 1; i < (k + 1) / 2; i++) { ans += l[i] * l[k-i]; } if (k % 2 == 0) { ans += l[k / 2] * (l[k / 2] - 1) / 2; } cout << ans << endl; return 0; }