给n个石头,每个石头都有一个权值,第i个石头的权值就是i,问把石头分成k堆,能不能使每一堆的权值和相等
看别人博客写了个不知名的定理,只要n个石头的权值和能整除k,就一定可以分出来,那就分类讨论了:1 当 n=偶 && n/k=奇 这种时候权值和是不能整除k的,2 当 n=偶 && n/k=偶 这种时候是最简单的分k堆,即n个数,分k行,每行偶数个数,1和n 2和n-1 … 就分完了 3当 n=奇 时,n/k也一定为奇,这种比较考验思维,借助上面的第二点,奇数个数,分成k行,每行奇数个数,那当每行只分一个数的时候,明显只有 n=k=1 才能成功,其他都是no 当每行分3个数以上的奇数个数时,我们分开来考虑,假如一行分7个数,分9行,那么把他想成7=4+3 意思就是每行提3个数出来单独分,前面四个数就像第二点那样塞进去,不过不是从1开始塞,而是从28开始塞,28和63组合,19和62组合,先把每行前4个塞满了,那就留下27个数了,问题就变成27个数分9行,没行3个怎么分,有一个构造方法:
1 2 3 4 5 6 7 8 9
5 6 7 8 9 1 2 3 4
9 7 5 3 1 8 6 4 2
讲解一下这个构造,我行列颠倒过来写,这样规律比较清晰,第一行按顺序1到9,第二行的第一个必须是5以上的,因为5是最小能保证1+5+9能到达15的数,如果是4,那么这一列到不了平均值15,为啥是15呢,这就简单到不用说了,然后确定了5,发现第二行第一个数一定是(1+k)/2,然后就5 6 7 8 9,到9之后再从1开始写到4,最后一行怎么确定呢,就用15减去一二行对应列下的数嘛,那为啥要这么写呢,第一行就是按顺序,习惯性写法,到第二行就要保证第二行写完之后,第一行每列加第二行都要产生一个不同的数字,因为这样第三行才能把1到9都填完,否则一二行之和出现相同数字,那么第三行是没有相同数字填的
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<time.h>
#include<math.h>
using namespace std;
#define ll long long
const int maxn=1e5+100;
int n,k,p;
void solve1()
{
printf("yes\n");
char c;
int cnt=1;
for(int i=1;i<=k;i++)
{
for(int j=1;j<=p/2;j++)
{
printf("%d %d",cnt,n-cnt+1);
if(j==p/2)printf("\n");
else printf(" ");
cnt++;
}
}
}
void solve2()
{
printf("yes\n");
char c=' ';
int cnt=3*k+1,cnt2=(1+k)/2,cnt3=0;
for(int i=1;i<=k;i++)
{
for(int j=1;j<=(p-3)/2;j++)
{
printf("%d %d ",cnt,n-cnt3);
cnt++;
cnt3++;
}
if((i-1)*2>k)cnt2=cnt2%k;
printf("%d %d %d\n",i,cnt2+k,(k+1)*3/2-i-cnt2+k+k);
cnt2++;
}
}
int main()
{
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
// clock_t st = clock();
int t;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&k);
p=n/k;
if(n==1 && k==1){
printf("yes\n1\n");
continue;
}
else if(k==n){
printf("no\n");
continue;
}
else if(k==1){
printf("yes\n");
for(int i=1;i<n;i++)printf("%d ",i);
printf("%d\n",n);
continue;
}
if(p%2==0 && n%2==0){
solve1();
}
else if(n%2==1){
solve2();
}
else{
printf("no\n");
}
}
// printf("time: %.0lfms\n", (clock()-st)*1000.0 / CLOCKS_PER_SEC);
return 0;
}