一道构造题

题意:

对于一个长度为$n$的01串，下标从0到$n-1$。定义两种类型的操作：
A类型：选择一个$x$将序列循环右移$x$位，也就是新序列的第$(i+x)\mathrm{m}\mathrm{o}\mathrm{d}n$位对应原序列的第$i$位。
例如：对于$x=2$,01100将会变成10001。
B类型：选择一个$x$,满足序列的第$x$个位置为1，且第$(x+1)\mathrm{m}\mathrm{o}\mathrm{d}n$位置不为1，交换序列的第$x$个位置和第$(x+1)\mathrm{m}\mathrm{o}\mathrm{d}n$个位置的字符。
对于01011,$x=0$和$x=3$均是不合法的，而$x=4$是合法的，操作后序列变为11010。
给定 $l$。请构造一个由$l$个长度为$n$的01串构成的序列$s[]$,下标从0开始.使得序列中每个串恰好有$k$个1。且对于$0<=i<l$,$s[i]$既可以通过一种A类型的操作，也可以通过一种B类型的操作，变为$s[i+1]$.
举个例子,$001$既可以通过一种A类型的操作也可以通过一种B类型的操作变为$100$:对于A类型可以设$x=1$,对于B类型可以设$x=2$.
输入格式
本题有多组数据，第一行一个数$t$表示数据组数。
接下来$t$行，每行三个正整数$n,k,l$.

解法

构造,考虑把01序列变为记录每个1之前有多少个0的序列,这样序列长度从$n$变为$k$,
然后对于新序列,A类型操作是要求序列循环同构,B类型操作相当于把某个数-1,它后面的数+1,然后如果执行了B操作之后还能执行A操作,说明序列元素差不会超过1.然后可以把所有元素减去$n/k$,然后变为01序列,变成了子问题.
无解情况:$gcd(n,k)!=1$

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	char c=getchar();int t=0,f=1;
	while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
	while(isdigit(c)){t=(t<<3)+(t<<1)+(c^48);c=getchar();}
	return t*f;
}
int t,n,k,l;
int p[105][105];
inline int gcd(int a,int b){
	if(b==0)return a;
	return gcd(b,a%b);
}
int q[105],pos;
void dfs(int n,int k){
	if(k==1){
		for(int i=1;i<=l;i++){
			p[i][i%n]=1;
		}
		return ;
	}
	dfs(k,n%k);
	int tmp=n/k;
	for(int j=0;j<k;j++)q[j]=p[1][j];
	int pre=0;
	for(int j=0;j<k;j++){
		for(int v=pre;v<=pre+tmp+q[j]-1;v++){
			if(v==pre+tmp+q[j]-1)p[1][v]=1;
			else p[1][v]=0;
		}
		pre+=tmp+q[j];
	}	
	for(int i=1;i<l;i++){int tmp=1;
		for(;tmp<n;tmp++){
			for(int j=0;j<n;j++)p[i+1][j]=p[i][(j+tmp)%n];
			int flag=1,pos1=0,pos2=0;
			for(int j=0;j<n;j++){
				if(p[i+1][j]!=p[i][j]){
					flag--;
					if(flag==0)pos1=j;
					if(flag==-1)pos2=j;
				}
				if(flag==-2)break;
			}
			if(flag==-1){
				if((p[i+1][pos1]==0)&&(p[i+1][pos2]==1)&&(pos2-pos1==1))break;
				if((p[i+1][pos1]==1)&&(p[i+1][pos2]==0)&&(pos1==0)&&(pos2==n-1))break;
			}
		}
	}
}
int main(){
	//freopen("a.in","r",stdin);
	//freopen("a.out","w",stdout);
	t=read();
	while(t--){
		memset(p,0,sizeof(p));
		n=read(),k=read(),l=read();
		if(gcd(n,k)!=1){
			puts("NO");
			continue;
		}
		dfs(n,k);
		puts("YES");
		for(int i=1;i<=l;i++){
			for(int j=0;j<n;j++)
			printf("%d",p[i][j]);
			puts("");
		}
	}
	return 0;
}