B. Two Out of Three

纯粹的發表於2024-07-21

原題連結

題解

假如 1 成立,23 不成立,代表對於 所有相等 的 \(a_i\) ,存在 \(b_i=1,b_i=2\) ,但是不存在 \(b_i=3\)

所以找出兩個不同的、出現次數大於1的數,然後這兩組分別附上 12,23,其他數賦值1,因為不會產生任何情況123

實施

首先找出兩個數,然後先給所有數賦1,再給其中一個各賦23

code

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

int a[105],b[105];

void solve()
{
    int n;
    cin>>n;

    map<int,int> q;
    set<int> st;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        int x=a[i];
        q[x]++;
        if(q[x]>1&&st.size()<2) st.insert(x);
    }
    if(st.size()>1)
    {
        for(int i=1;i<=n;i++) b[i]=1;
        for(int i=1;i<=n;i++)
        {
            if(a[i]==*st.begin())
            {
                b[i]=2;
                break;
            }
        }
        for(int i=1;i<=n;i++)
        {
            if(a[i]==*st.rbegin())
            {
                b[i]=3;
                break;
            }
        }
        for(int i=1;i<=n;i++) cout<<b[i]<<' ';
        cout<<'\n';
    }
    else cout<<"-1\n";
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t=1;
    cin>>t;
    while(t--) solve();
    return 0;
}