CF1004C Sonya and Robots(简单计数+STL)

CF1004C Sonya and Robots

题意：

给你$n$个数，从每一个数起向后找一个数，两个数组成一个数对，求$n$个数能有多少个不同的数对。
$1\le n\le 1e5$

思路：

对于计数需要去重。
假设数组$s{z}_i$表示第$i+1\sim n$个数中有多少个不同的数。
从前往后遍历

• 若一个数$x$还没出现过，则$x$对答案的贡献是$s{z}_i$
• 若一个数$x$在之前出现过了，则$x$对答案的贡献是$0$，因为后面的数对在前面的$x$中已经计算过了。

$sz$数组可以用$STL$中的$set$逆序遍历求得。

Code:

int q[N];
int sz[N];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    cin>>n;
    rep(i,1,n){
        cin>>q[i];
    }
    set<int>Q;
    map<int,int>mp;
    ll ans=0;
    dep(i,n,1){
        sz[i]=Q.size();//不同数的个数
        Q.insert(q[i]);
    }   
    rep(i,1,n){
        if(mp[q[i]]==0){
            ans+=sz[i];//第一次出现
        }
        mp[q[i]]++;
    }
    cout<<ans;
    return 0;
}