Codeforces Round #495 (Div. 2)-C. Sonya and Robots

题意:题目描述其实蛮复杂,反正就是找给出的序列里有多少组不同的<ai,aj>(i<j),a[i]和a[j]可以相同。

题解:超级简单,只需要从后往前统计到每个位置为止有多少种不同的数,最后再从前往后遍历一次,对于每个不同的ai只需要找后面有多少个不同的bi就行了,也就是加上他后面不同的数的个数。

附上代码:

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;


const int maxn=100010;
int a[maxn];
int cnt[maxn];
int vis[maxn];


int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);//输入每个数
    memset(vis,0,sizeof(vis));
    memset(cnt,0,sizeof(cnt));
    for(int i=n; i>=1; i--) //倒着统计每个位置有多少种的数
    {
        cnt[i]=cnt[i+1];
        if( !vis[a[i]] )
        {
            vis[a[i]]=1;
            cnt[i]++;
        }
    }
    memset(vis,0,sizeof(vis));
    ll sum=0;
    for(int i=1; i<=n; i++)
    {
        if( !vis[a[i]] )
        {
            sum += cnt[i+1];
            vis[a[i]]=1;
        }
    }
    printf("%lld\n",sum);
}

猜你喜欢

转载自blog.csdn.net/wookaikaiko/article/details/80959470