Codeforces Round#770(Div.2) C. OKEA

题目

你为一家知名百货公司工作，该百货公司使用领先的技术并采用机械工作——即机器人！

你工作的部门销售 n⋅k 件商品。第一件物品花费 1 美元，第二件物品花费 2 美元，依此类推：第 i 件物品花费 i 美元。这些物品位于货架上。这些物品形成一个矩形网格：总共有 n 个架子，每个架子正好包含 k 个物品。我们将用 ai,j 表示第 i 个货架上第 j 件商品（从左数起）的价格，1≤i≤n,1≤j≤k。

有时机器人会好奇并思考以下问题：对于某些货架 i 和索引 l≤r，物品$a_{i,l},a_{i,l+1},...,a_{i,r}$ 的平均价格（算术平均值）是多少？不幸的是，旧机器人只能处理整数。如果平均价格不是整数，它们就会崩溃。

你关心机器人的福利。您希望以机器人理论上不会损坏的方式排列物品。形式上，您要选择这样一个二维数组 a：

从 1 到 n⋅k（包括）的每个数字只出现一次。
对于每个 i,l,r，第 i 个货架上从 l 到 r 的商品的平均价格是一个整数。
找出这种安排是否可行，如果可行，请举出这种安排的任何例子。

输入

第一行包含一个整数 t (1≤t≤500)——测试用例的数量。

每个测试用例的第一行也是唯一一行包含两个整数 n 和 k (1≤n,k≤500)——分别是货架的数量和每个货架的长度。

保证所有测试用例的和 n 不超过 500，所有测试用例的和 k 不超过 500。

输出

打印每个测试用例的答案。

如果存在这样的安排，请在一行上打印“YES”。之后，在每行 k 个数字的 n 行上打印任何示例，每个书架一行。从 1 到 n⋅k 的每个数字必须在输出中恰好出现一次。

如果不存在良好的安排，请在其单独的行上打印一个单词“NO”。

题解

如果 k=1，可以按任意顺序将物品放在货架上。

否则，每个架子上至少有 2 件物品。如果货架上有奇偶性不同的物品，很明显有两个相邻的奇偶性不同的物品，两个物品的算术平均值不会是整数，因此，每个货架上的所有物品都是同奇偶性的。

请注意，如果货架数 n 为奇数，我们将无法正确排列物品，因为包含偶数和奇数物品的货架数必须相同（即，如果 k≥2）。

让我们证明，即使是 n，也总有一个答案。在第 i 个货架上，我们将放置价格为 i,i+n,i+2⋅n,…,i+n⋅(k−1) 的商品。我们可以使用算术级数之和的公式来计算坐标为 i,l 到 i,r 的子段的价格之和：

sum=i⋅(r−l+1)+n(l−1)+n(r−1)/2⋅(r−l+1)= =i⋅(r−l+1)+n/2⋅(l +r−2)⋅(r−l+1)= =(r−l+1)⋅(i+n/2⋅(l+r−2))
段的长度 (r-l+1) 总是除此总和，因为 n 是偶数。因此，这种安排符合问题的要求。
那我们构造公差为2的等差数列就可以将所有数放到货柜上面了。

#include<bits/stdc++.h>
using namespace std;
int mp[505][505],n,m,st[30005];
int solve()
{
	cin >> n >> m;
	if (n % 2 == 1&&m!=1)return 0;
	int x = 1;
	if (m == 1) {
		for (int i = 1; i <= n; i++)for (int j = 1; j <= m; j++)mp[i][j] = x++;
		return 1;
	}
	int temp = 1;
	for (int i = 1; i <= n / 2; i++)
	{
		for (int j = 1; j <= m; j++)
			mp[i][j] = temp, temp += 2;
	}
	temp = 2;
	for (int i = n/2+1; i <= n; i++)
	{
		for (int j = 1; j <= m; j++)
			mp[i][j] = temp, temp += 2;
	}
	return 1;

}
int main()
{
	int t;
	cin >> t;
	while (t--)
	{   
		if (solve())
		{
			cout << "YES" << endl;
			for (int i = 1; i <= n; i++)
			{
				for (int j = 1; j <= m; j++)
				{
					cout << mp[i][j] << " ";
				}
				cout << endl;
			}
		}
		else cout << "NO" << endl;
	}
}