枚举算法-大力能否出奇迹 Problem F. 课堂作业-7-3 数组配对问题

Problem F. 课堂作业-7-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<bits/stdc++.h>
using namespace std;
int a[1000005];
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++){
		int x;
		scanf("%d",&x);
		a[x%k]++;
	}
	long long ans=0;
	for(int i=0;i<k;i++){
		int j=(k-i)%k;//注意点1 
		if(j<i)break;//防止重复配对 
		if(j==i){//注意点2 
			ans+=(long long)a[i]*(a[i]-1)/2;
		}
		else{
			ans+=(long long)a[i]*a[j];
		}
	}
	printf("%lld",ans);
	return 0;
}

注意点1：j=(k-i)%k

这个很好理解，举例：对于一个数组a[5]={1,2,3,4,5}

i=0 1 2 3 4 5，那么j就等于:

j=0 5 4 3 2 1

其中0和k/2为边界情况，这也是注意点2。

在边界情况即i=j=0或i=j=k/2时，问题转化为组合问题 ，即从n个数中取2个数。

同时需要注意对ans的结果强制转换类型long long，因为可能会爆掉int