codeforces B. Two Out of Three

记录篇

分析

  • 改题比较简单 想要满足任何一种情况首先你需要 a a a 存在相等的两个元素,此时你任意选择一种情况即可 1 , 2      1 , 3      2 , 3 1,2\;\;1,3\;\;2,3 1,21,32,3 ,仔细想你可以发现,如果该元素的数量大于两个,你也只能够选择一种情况,也就是说你不能够同时填 1    2    3 1\;2\;3 123 (因为这样三种情况同时存在)。

结论

  • 通过分析易得:任意有重复数量 ≥ 2 \ge2 2 的可以满足一种情况,所以至少需要两种元素,且这两种元素重复数量至少为 2 2 2

思路

  • 遍历计数每个元素,随便选两个计数数量大于等于 2 2 2 的。随便分配这两种元素一种情况看你心情 。别的没选中的元素一律填一样的数字。
    Think Twice, Code Once
#include
#define il inline
#define get getchar
#define put putchar
#define is isdigit
#define re register
#define int long long
#define dfor(i,a,b) for(re int i=a;i<=b;++i)
#define dforr(i,a,b) for(re int i=a;i>=b;--i)
#define dforn(i,a,b) for(re int i=a;i<=b;++i,put(10))
#define mem(a,b) memset(a,b,sizeof a)
#define memc(a,b) memcpy(a,b,sizeof a)
#define pr 114514191981
#define gg(a) cout<<a,put(32)
#define INF 0x7fffffff
#define tt(x) cout<<x<<'\n'
#define ls i<<1
#define rs i<<1|1
#define la(r) tr[r].ch[0]
#define ra(r) tr[r].ch[1]
#define lowbit(x) (x&-x)
using namespace std;
typedef unsigned int ull;
int read(void)
{
    re int x=0,f=1;re char c=get();
    while(!is(c)) (f=c==45?-1:1),c=get();
    while(is(c)) x=(x<<1)+(x<<3)+(c^48),c=get();
    return x*f;
}
void write(int x)
{
    if(x<0) x=-x,put(45);
    if(x>9) write(x/10);
    put((x%10)^48);
}
#define writeln(a) write(a),put(10)
#define writesp(a) write(a),put(32)
#define writessp(a) put(32),write(a)
const int N=5e6+10,M=3e4+10,SN=1e4+10,mod=998244353;
int n,a[SN],cnt[SN];
bool vis[SN];
signed main()
{
    int T=read();
    while(T--)
    {
        n=read();
        vector<int > ve;
        mem(cnt,0),mem(vis,0);
        dfor(i,1,n) a[i]=read(),++cnt[a[i]];
        re int sum=0;
        for(re int i=1;i<=n&&sum<2;++i)
            if(cnt[a[i]]>1&&!vis[a[i]]) ve.push_back(a[i]),++sum,vis[a[i]]=1;
        if(sum<2){puts("-1");continue;}
        re bool f1=0,f2=0;
        dfor(i,1,n)
        {
            if(a[i]==ve[0])
            {
                if(!f1) writesp(1),f1=1;
                else writesp(2);
            }
            else if(a[i]==ve[1])
            {
                if(!f2) writesp(1),f2=1;
                else writesp(3);
            }
            else writesp(2);
        }
        puts("");
    }
    return 0;
}

你可能感兴趣的:(codeforces题解,算法,数据结构,codeforce题解)