Codeforces 791C Bear and Different Names 简单构造

点击打开链接

题意:给出1~n-k+1连续k个人的名字是相同或者不同,构造出满足条件的n个名字 n,k<=50
任意生成50个可重排列,则a,aa,a..a,b,bb,b...b 前5个字符每种各10个 
简单构造:若i~i+k-1为N,则令ans[i]=ans[i+k-1],使得i+1~i+k-1为Y,则i+1~i+k为N还是Y只取决于ans[i+k] 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e2+20;
char s[N][N];
int n,k;
string a[N];
int ans[N];//ans[i],保存第i个名字在s的id 
void init()
{
	int i=0,b=0;
	while(i<50)
	{
		int l=1;
		while(l<=5)	
		{
			++i;
			for(int k=1;k<=l;k++)
			{
				if(k==1)
				s[i][k]='A'+b;
				else
				s[i][k]='a'+b;	
			}
			s[i][l+1]='\0';
			l++;
		}
		b++;
	}
}
int main()
{
	init();
	while(cin>>n>>k)
	{
		for(int i=1;i<=n-k+1;i++)
			cin>>a[i];
		int x=0;//当前编号
		
		
		
		if(a[1]=="YES")
		{
			for(int i=1;i<=k;i++)
				ans[i]=++x;	
		}
		if(a[1]=="NO")
		{
			for(int i=1;i<=k;i++)
			{
				if(i!=k)
				ans[i]=++x;
				else
				ans[i]=ans[1];
			}
		}
		
		for(int i=2;i<=n-k+1;i++)
		{
			if(a[i]=="NO")
				ans[i+k-1]=ans[i];	
			else
				ans[i+k-1]=++x;	
		}
		for(int i=1;i<=n;i++)
			printf("%s ",s[ans[i]]+1);
		printf("\n");	
	}
	return 0;
}