C. Fence Painting

显然对于我们不同的就直接修改
但是我们应该贪心的叫更后面来的人来修改
这样前面的人怎么造都无所谓了

for(int i=1;i<=n;i++){
        if(a[i]!=b[i]){
            if(cnt[b[i]].size()){
                auto t=cnt[b[i]].back();
                if(b[i]==c[m]&&!flag)ypos=i,flag=1;
                ans[t]=i;
                cnt[b[i]].pop_back(),a[i]=b[i];
            }else{NO return;}
        }
    }

然后我们就只需要判断最后一个人 他是否被用了
要是被用了 我们其他改变的就直接随过来
否则 我们应该知道最后一个人他在b内是否有对应的 如果没有的话
显然错误的

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
const int M = 998244353;
const int mod = 1e9+7;
#define int long long
int up(int a,int b){return a<0?a/b:(a+b-1)/b;}
#define endl '\n'
#define all(x) (x).begin(),(x).end()
#define YES cout<<"YES"<<endl;
#define NO cout<<"NO"<<endl;
#define _ 0
#define pi acos(-1)
#define INF 0x3f3f3f3f3f3f3f3f
#define fast ios::sync_with_stdio(false);cin.tie(nullptr);

void solve(){
    int n,m;cin>>n>>m;
    vector<int>a(n+10),b(n+10),c(m+10);
    vector<int>cnt[n+10];
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)cin>>b[i];
    for(int i=1;i<=m;i++){
        cin>>c[i];
        cnt[c[i]].push_back(i);
    }
    vector<int>ans(m+1);
    int y=cnt[c[m]].size();
    int ypos,flag=0;
    for(int i=1;i<=n;i++){
        if(a[i]!=b[i]){
            if(cnt[b[i]].size()){
                auto t=cnt[b[i]].back();
                if(b[i]==c[m]&&!flag)ypos=i,flag=1;
                ans[t]=i;
                cnt[b[i]].pop_back(),a[i]=b[i];
            }else{NO return;}
        }
    }
    if(y==cnt[c[m]].size()){
        if(std::count(b.begin(), b.end(), c[m])){
            YES
            int pos;
            for(int j=1;j<=n;j++)if(b[j]==c[m])pos=j;
            for(int j=1;j<=m;j++)
                if(!ans[j])
                    ans[j]=pos;
            for(int j=1;j<=m;j++)cout<<ans[j]<<' ';cout<<endl;
        }else NO
    }else{
        YES
        for(int j=1;j<=m;j++)
            if(!ans[j])
                ans[j]=ypos;
        for(int j=1;j<=m;j++)cout<<ans[j]<<' ';cout<<endl;
    }
}
signed main(){
    fast
    int t;t=1;cin>>t;
    while(t--) {
        solve();
    }
    return ~~(0^_^0);
}

原文地址：http://www.cnblogs.com/ycllz/p/16807978.html