https://codingcompetitions.withgoogle.com/kickstart/round/000000000019ff47/00000000003bef73
特判一下1 2的情况
然后如果我们需要A-C个2放在左边给左边的人看,B-C个2放右边给右边的人看,C个n放在中间给两个人都看,剩下不足的位置就放1,但是1一定要被挡住,所以C>2时1就放在C中间,如果A-C>0就放在左边的2和n中间,B-C>0就放在右边的n和2中间,否则则无解
#include<bits/stdc++.h>
#define pb push_back
using namespace std;
typedef long long ll;
const int maxl=3e5+10;
int n,m,cas,k,cnt,tot,ans;
int A,B,C;
int a[maxl],b[maxl];
char s[maxl];
bool in[maxl];
inline void prework()
{
scanf("%d%d%d%d",&n,&A,&B,&C);
}
inline void mainwork()
{
ans=0;
if(A-C+B-C+C>n || C>n || C<=0)
{
ans=0;
return;
}
if(n==1)
{
if(A!=1 || B!=1 || C!=1)
{
ans=0;return;
}
ans=1;a[1]=1;return;
}
if(n==2)
{
if(C==2 && A==2 && B==2)
{
ans=1;a[1]=1;a[2]=1;
}
if(C==1 && A==2 && B==1)
{
ans=1;a[1]=1;a[2]=2;
}
if(C==1 && A==1 && B==2)
{
ans=1;a[1]=2;a[2]=1;
}
return;
}
cnt=0;
if(C>=2)
{
ans=1;
for(int i=1;i<=A-C;i++)
a[++cnt]=2;
a[++cnt]=n;
for(int i=1;i<=n-(A-C+B-C+C);i++)
a[++cnt]=1;
for(int i=2;i<=C;i++)
a[++cnt]=n;
for(int i=1;i<=B-C;i++)
a[++cnt]=2;
}
else if(A-C>0)
{
ans=1;
for(int i=1;i<=A-C;i++)
a[++cnt]=2;
for(int i=1;i<=n-(A-C+B-C+C);i++)
a[++cnt]=1;
for(int i=1;i<=C;i++)
a[++cnt]=n;
for(int i=1;i<=B-C;i++)
a[++cnt]=2;
}
else if(B-C>0)
{
ans=1;
for(int i=1;i<=A-C;i++)
a[++cnt]=2;
for(int i=1;i<=C;i++)
a[++cnt]=n;
for(int i=1;i<=n-(A-C+B-C+C);i++)
a[++cnt]=1;
for(int i=1;i<=B-C;i++)
a[++cnt]=2;
}
else
ans=0;
}
inline void print()
{
printf("Case #%d:",cas);
if(!ans)
puts(" IMPOSSIBLE");
else
{
for(int i=1;i<=n;i++)
printf(" %d",a[i]);
puts("");
}
}
int main()
{
int t=1;
scanf("%d",&t);
for(cas=1;cas<=t;cas++)
{
prework();
mainwork();
print();
}
return 0;
}