Educational Codeforces Round 51 (Rated for Div. 2) C. Vasya and Multisets

题目大意就是给你n个数,然后问你能否把这n个数分给2个集合,使得2个集合里的nice number个数相同,(nice number即在该集合里仅出现一次的数)。问是否存在分配方案,首先 一共有三种情况,如果n个数里仅出现一次的数的个数为偶数个,那么肯定存在。还有就是如果仅出现一次的数的个数为奇数个,那么还要看是否存在出现次数>=3的数,有的话就存在,没有的话就不存在。

#include<bits/stdc++.h>
using namespace std;
int vis[105];
vector<int>v[105];
int main()
{
    int n,cnt1=0,cnt3=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int tmp;
        scanf("%d",&tmp);
        v[tmp].push_back(i);
    }
    for(int i=1;i<=100;i++)
    {
        if(v[i].size()==0)
            continue;
        if(v[i].size()==1)
            cnt1++;
        else
            if(v[i].size()>=3)
                cnt3++;
    }
    if(cnt1%2==0)
    {
        int tmp=0;
        for(int i=1;i<=100;i++)
        {
            if(v[i].size()==0)
                continue;
            if(v[i].size()==1)
            {
                if(tmp<cnt1/2)
                    vis[v[i][0]]=1,tmp++;
                else
                    vis[v[i][0]]=2;
            }
            else
                for(int j=0;j<v[i].size();j++)
                    vis[v[i][j]]=1;
        }
    }
    else
    {
        int flag=1;
        if(cnt3==0)
        {
            cout<<"NO"<<endl;
            return 0;
        }
        int tmp=0;
        for(int i=1;i<=100;i++)
        {
            if(v[i].size()==0)
                continue;
            if(v[i].size()==1)
            {
                if(tmp<(cnt1/2+1))
                    vis[v[i][0]]=1,tmp++;
                else
                    vis[v[i][0]]=2;
            }
            else
                if(v[i].size()==2)
                    vis[v[i][0]]=vis[v[i][1]]=1;
                else
                {
                    if(flag)
                    {
                        vis[v[i][0]]=2;
                        for(int j=1;j<v[i].size();j++)
                            vis[v[i][j]]=1;
                            flag=0;
                    }
                    else
                        for(int j=0;j<v[i].size();j++)
                            vis[v[i][j]]=1;
                }
        }
    }
    cout<<"YES"<<endl;
    for(int i=1;i<=n;i++)
        if(vis[i]==1)
            cout<<"A";
        else
            cout<<"B";
    cout<<endl;
    return 0;
}

 

posted @ 2018-10-30 18:55  eason99  阅读(76)  评论(0编辑  收藏  举报