Submission #42209134

---

Source Code Expand

#include<cstdio>
#include<cstring>
#define fep(i,x,y) for(int i=x;i<=y;i++)
#define feq(i,x,y) for(int i=x;i>=y;i--)
#define N 60
#define M 11000
using namespace std;
inline int mymin(int x,int y){return x<y?x:y;}
inline int mymax(int x,int y){return x>y?x:y;}
inline void zwap(int &x,int &y){x^=y^=x^=y;}
int a[N],c[N],n,m;
int b[M][N],top;
int s1[N],t1,s2[N],t2;
int list[N],tail;
inline void chuli(int x,int y,int k){
	tail=1;int tx=1,ty=1+k;list[1]=x;list[ty]=y;
	if(ty==2)tail++;
	fep(i,1,n){
		if(i!=x && i!=y)list[++tail]=i;
		if(tail==ty-1)tail++;
	}
	top+=2;
	fep(i,1,n)b[top-1][list[i]]=i,b[top][list[i]]=n-i+1;
	zwap(b[top][x],b[top][y]);
	a[x]-=k;a[y]+=k;
}
int main(){
	scanf("%d",&n);
	int cnt=0;
	fep(i,1,n){
		scanf("%d",&a[i]);
		cnt+=a[i];
	}
	if(n&1)m=n;
	else if(!(n&1)){
		m=n/2;int ji=-n/2;
		if(cnt%n!=0){
			fep(i,1,n)c[i]=ji+i;
		}
	}
	if(cnt%m!=0){
		printf("No\n");
		return 0;
	}
	int k=cnt/n;
	fep(i,1,n)a[i]-=k+c[i];
	fep(i,1,n){
		if(a[i]>0)s1[++t1]=i;
		else if(a[i]<0)s2[++t2]=i;
	}
	while(t1){
		int k=mymin(mymin(a[s1[t1]],-a[s2[t2]]),n-1);
		chuli(s1[t1],s2[t2],k);
		if(!a[s1[t1]])t1--;
		if(!a[s2[t2]])t2--;
	}
	if(!(n&1) && cnt%n){
		top++;
		fep(i,1,n)b[top][i]=n-i+1;
	}
	printf("Yes\n");
	printf("%d\n",top);
	fep(i,1,top){
		fep(j,1,n)printf("%d ",b[i][j]);
		printf("\n");
	}
	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘void chuli(int, int, int)’:
./Main.cpp:16:13: warning: unused variable ‘tx’ [-Wunused-variable]
   16 |  tail=1;int tx=1,ty=1+k;list[1]=x;list[ty]=y;
      |             ^~
./Main.cpp: In function ‘int main()’:
./Main.cpp:28:7: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   28 |  scanf("%d",&n);
      |  ~~~~~^~~~~~~~~
./Main.cpp:31:8: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   31 |   scanf("%d",&a[i]);
      |   ~~~~~^~~~~~~~~~~~

Judge Result