原文链接https://www.cnblogs.com/zhouzhendong/p/AGC030C.html

题解

　　才发现当时是被题意杀了。

　　当时理解的题意是“对于任意的 (i,j) ，颜色 i 和 颜色 j 的相邻格子对数相同”。好像当时看了样例都没有发现被题意杀……

　　回归正题。

　　如果要构造 n = k 的，那么直接 构造一个第 i 行为 i 的就好了。

　　那么 n<k<=2n 呢？

　　我们考虑斜着的列。

　　如果第 i 个斜列放 i ，那么也可以达到 n = k 的目的。

　　如果第 i 个斜列间隔放两种数，那么仍然可以满足条件。这时候就可以做到 $k\in [n,2n]$ 了。

　　放个图便于理解。（下图蓝线穿过的列是特殊的列）

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
LL read(){
	LL x=0;
	char ch=getchar();
	while (!isdigit(ch))
		ch=getchar();
	while (isdigit(ch))
		x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return x;
}
const int N=505;
int n,k;
int g[N][N];
int main(){
	k=read();
	n=min(k,500);
	for (int i=1;i<=n;i++){
		int x=1,y=i;
		for (int j=1;j<=n;j++){
			g[x][y]=i;
			x=x%n+1,y=y%n+1;
		}
	}
	for (int i=n+1;i<=k;i++){
		int x=1,y=i-n;
		for (int j=1;j<=n;j++){
			if (j&1)
				g[x][y]=i;
			x=x%n+1,y=y%n+1;
		}
	}
	printf("%d\n",n);
	for (int i=1;i<=n;i++,puts(""))
		for (int j=1;j<=n;j++)
			printf("%d ",g[i][j]);
	return 0;
}

原文地址：https://www.cnblogs.com/zhouzhendong/p/AGC030C.html