Codeforces1264 B. Beautiful Sequence(枚举)

题意:

有a个0,b个1,c个2,d个3
要求用这些数构造一个长度为a+b+c+d的序列,满足相邻数的差值为1。
如果无解输出NO,否则输出YES和一种合法解

数据范围:a+b+c+d<=1e5

解法:

暴力枚举起点k,设当前指针为now=k,
每加入队列一次就判断now-1是否还有数,如果有就转移到now-1,否则转移到now+1

指针转移方向无所谓的,优先左边或者优先右边都可以,
因为如果有解的话:
假设优先左边,左边构造完,now肯定会回到k,只有这样才能继续向k+1转移
假设优先右边,右边构造完,now肯定会回到k,只有这样才能继续向k-1转移

但是如果优先左边就得把左边都用完才能回来,否则显然会出错

code:

#include<bits/stdc++.h>
using namespace std;
const int maxm=3e5+5;
int a[4];
int b[4];
signed main(){
    for(int i=0;i<4;i++)cin>>a[i];
    vector<int>ans;
    for(int k=0;k<4;k++){//枚举起点
        if(!a[k])continue;
        for(int i=0;i<4;i++)b[i]=a[i];
        vector<int>temp;
        int now=k;
        while(1){
            temp.push_back(now);
            b[now]--;
            if(now&&b[now-1]){
                now--;
            }else if(now!=3&&b[now+1]){
                now++;
            }else{
                break;
            }
        }
        if(b[0]+b[1]+b[2]+b[3]==0){
            ans=temp;break;
        }
    }
    if(!ans.empty()){
        cout<<"YES"<<endl;
        for(int v:ans){
            cout<<v<<' ';
        }
    }else{
        cout<<"NO"<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_44178736/article/details/107792867