C. Sonya and Robots codeforces 1004

Posted 2021-02-05 lishengkangshidatiancai

解题思路：把每个数值都记录下来，并且记录共有多少中不同的数值，然后从左向右走，每到一个数是值这个数记录下，以保证下次不会再记录它，然后加上剩余剩的种类数。注意：结果要用long long存。这种暴力开map的方法屡试不爽

#include<bits/stdc++.h>
using namespace std;
int n;
int a[100010];
int book[100010];
map<int,int> mp;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
       scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
    {
        if(mp.find(a[i])!=mp.end())
        {
            mp[a[i]]++;
        }
        else
        {
            mp[a[i]]=1;
        }
    }
    long long ans=0;
    for(int i=1;i<=n;i++)
    {
        mp[a[i]]--;
        if(mp[a[i]]==0)
           mp.erase(a[i]);
        if(!book[a[i]])
        {
            ans+=mp.size();
            book[a[i]]=1;
        }
    }
    printf("%lld
",ans);
}

当然也可以用桶并且用一个整形记录数量，这样更快一点

#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<cstring>
#include<utility>
#define endl ‘
‘
#define _ ios::sync_with_stdio(false)
bool SUBMIT = 1;
typedef long long ll;
using namespace std;
const int inf = 1e5+100;
int n,k;
vector<int>g;
int s[inf],c[inf];
int main()
{
    if(!SUBMIT)freopen("i.txt","r",stdin);else _;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        int l;cin>>l;g.push_back(l);
        s[l]++;
        if(s[l]==1)k++;
    }
    ll ans=0;
    for(int i=0;i<n;i++)
    {
        s[g[i]]--;
        if(s[g[i]]==0)k--;

        if(!c[g[i]]){
            ans+=k;
            c[g[i]]=1;
        }
    }
    cout<<ans<<endl;
    return 0;
}