题意:
有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;
}