ABC206 C - Swappable（map计数）

Posted 2021-07-23 live4m

题意：

解法：

map统计每种数的数量,
对于数字x,设数量为cnt,那么!=x的数量为n-cnt,
则ans+=cnt*(n-cnt),
由于题目求的是无序对,最后答案需要除以2.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=4e6+5;
int a[maxm];
int n;
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    map<int,int>mp;
    for(int i=1;i<=n;i++){
        mp[a[i]]++;
    }
    int ans=0;
    for(auto i:mp){
        int x=i.second,y=n-x;
        ans+=x*y;
    }
    cout<<ans/2<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
    return 0;
}

---