武大Shopee杯-B

武大Shopee杯-B_第1张图片
武大Shopee杯-B_第2张图片
题目大意:
给定n个数,从n个数中选出两个数之和为0,为一种匹配方式,每个数不能与本身匹配,问总共可能的匹配数目。

解题思路:
由于数的范围不大,求出每个数出现的次数,则i+j==0,匹配方法为cnt[i]*cnt[j]种,注意0与0匹配的时候比较特殊。

#include 
#define ll long long
using namespace std;
const int maxn=5e5+10;
ll cnt[41];
ll ans;
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
    {
       int temp;
       cin>>temp;
       cnt[temp+20]++;
    }
    for(int i=0;i<=19;++i)
    ans+=cnt[i]*cnt[40-i];
    ans+=cnt[20]*(cnt[20]-1)/2;
    cout<<ans<<endl;
    return 0;
}

你可能感兴趣的:(数论)