题目描述->戳我
题意
t个数,求可以构成相反数的有多少对,比如(1,0,-1,0,1)这组可以构成3对。
果然和cp有关系就会上头然后出错呢
数据范围
n<=5*1e5,-20<=at<=20
样例
输入:
6
-1 1 0 0 1 0
输出:
5
思路
将数组平移,类似于桶排序的思想。每次输入a[i],对应的cnt[a[i]+20]++,然后将关于20对称的两个数两两相乘。
坑点
由于数量很多,最后求总数时候有可能达到1e10然后爆int,记得开longlong ( 改了半个小时的我疲惫的笑了出来)
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
long long a[500005],c[50]={0};
int main(){
int n;
memset(c,0,sizeof(c));
while(~scanf("%d",&n)){
for(int i = 0;i < n;i++){
scanf("%lld",&a[i]);
c[a[i]+20]++;
}
long long cnt = 0;
int i;
for( i = 0; i <= 40;i++){
if(i==20){
cnt=cnt+c[20]*(c[20]-1)/2;
c[20] = 0;
}
if(c[i]!=0&&c[40-i]!=0){
cnt=cnt+c[i]*c[40-i];
c[i] = 0;
c[40-i] = 0;
}
}
cout << cnt << endl;
}
return 0;
}
p.s.
即使题目没有要求多组输入 也尽量满足多组,好习惯++