k倍数

it2026-08-24  3

时间限制 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; }
最新回复(0)