Codeforces Round #558 (Div. 2)-Cat Party (Hard Edition)-（前缀和 + 模拟）

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<vector>
#include<iostream>
#include<map>
#include<cstring>
#include<set>
#include<queue>
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;

int a[200086];///每一天的颜色
int color[200086];///下标颜色出现的次数
int n;


int main()///cf558div2B
{
    while(scanf("%d",&n)!=EOF)
    {
        memset(a,0,sizeof(a));
        memset(color,0,sizeof(color));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        int sum=0;/// 一共出现几种颜色
        int maxx=-1,ans=0;///最大次数 答案
        int one=0; ///特判颜色次数为1的颜色数量
        set<int>se;///判断当前颜色是否出现过
        map<int,int>mp;///快速找到最大次数的颜色数
        for(int i=1;i<=n;i++)
        {
            if( !se.count(a[i]) )
            {
                sum++;
                se.insert(a[i]);
            }
            color[ a[i] ]++;
            mp[ color[ a[i] ] ]++;///每一次都累加，最后只需要用到maxx和maxx-1

            if( color[ a[i] ]==1 )
                one++;
            else if( color[ a[i] ]==2 )///颜色出现的次数超过1自然会先变成2，并且回不来了
                one--;

            if( color[ a[i] ] > maxx )
                maxx=color[  a[i] ];

            if( mp[maxx]==1 && mp[maxx-1]==sum )
                ans=i;
            if( mp[maxx]==sum-1 && one==1 )
                ans=i;
            if(maxx==1 && mp[maxx]==sum)
                ans=i;
        }
        printf("%d\n",ans);
    }
    return 0;
}