4.12Shopee杯武汉大学网络预选赛 第二题

题目描述->戳我

题意
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.

即使题目没有要求多组输入 也尽量满足多组,好习惯++

你可能感兴趣的:(算法)