这题比赛的时候好像spj写挂了,所以导致我当时那个很挫的代码都过了,其实没那么简单。
首先,我们来讨论不可能的情况:
1.一共有n个数,那么1-n的总和 sum = (1+n)*n/2,如果sum不能整除m,那么输出NO;
2.令each = sum/m,这是每个组的总和。我们知道,1-n我们都得用,所以假设each<n,那么这也是无解的,输出NO;
接下来就是有解的情况,我们通过以下的方法来构造这m组和相同的数据。
假如此时n=50,m=5,那么可以这么构造:
55 46 45 36 35 26 25 16 15 9
54 47 44 37 34 27 24 17 14 10
53 48 43 38 33 28 23 18 13 11
52 49 42 39 32 29 22 19 12 8 4
51 50 41 40 31 30 21 20 7 6 5 3 2 1
我们发现,从第一列起,这个构造是呈蛇形分布的,即
第一列,从上到下,依次减小
第二列,从下到上,依次增大
第三列,从上到下,依次减小
。。。。。。。。。。。。。
但是我们发现,不是所有数据都能这么构造的,到某个值s的时候,我们就不能继续这么蛇形构造了,那么这个s是多少那?
我们知道,each=sum/m>=n, sum =(1+n)*n/2,代入得
n>=2*m-1,这就说明,如果要有解,那么n>=2*m-1。
那说明,我们这样蛇形构造的时候,如果不能全部构造完,那么一定要注意最后剩下的数字n得大于等于2*m-1。那么怎么才能保证这一点那?我们知道,构造一次需要的数字数目为2*m,那么当我们还剩小于2m+2*m-1=4*m-1时,我们一定要停止构造了。因为如果再构造一次,剩下的数目就小于2*m-1,由上面的分析可知,这个是无解的。
那剩下的4*m-1的数怎么构造?我们爆搜即可,这样,我们就把搜索的范围缩小到了<40。
整个这样下来,就可以构造出这m组数据了。
在此感谢博友的热情解释,要不然我还看不懂这个4*m的梗。
#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<time.h>
#include<queue>
#include<stack>
#include<iterator>
#include<math.h>
#include<stdlib.h>
#include<limits.h>
#include<map>
#include<set>
//#define ONLINE_JUDGE
#define eps 1e-8
#define INF 0x7fffffff
#define FOR(i,a) for((i)=0;i<(a);(i)++)
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,num) scanf("%d%d%d",&a,&b,&num)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
#define ll __int64
const double PI=acos(-1.0);
template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
template<class T> inline T Min(T a,T b){return a<b?a:b;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
using namespace std;
int n,m;
#define M 110
#define N 100010
#define Mod 1000000007
#define p(x,y) make_pair(x,y)
vector<int> v[15];
int f[15];
ll res;
int flag;
void dfs(int num){
if(num == 0){
flag=1;
return;
}
if(flag)
return;
for(int i=0;i<m;i++){
if(f[i]+num>res) continue; //res为现在每个组的和
f[i] += num;
v[i].push_back(num);
dfs(num-1);
if(flag)
return;
f[i] -= num;
v[i].pop_back();
}
}
int main() {
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
#endif
int t;
sf(t);
while(t--){
sfd(n,m);
for(int i=0;i<=m;i++) v[i].clear();
ll sum = (ll)(1+n)*n/2;
ll each = sum/m;
if(sum%m || each<n){ //无法构造的情况
printf("NO\n");
continue;
}
int tmp=n;
for(int i=n;i>=4*m-1;){ //蛇形构造>=4*m-1的数据
for(int j=0;j<m;j++){
v[j].push_back(i);
i--;
}
for(int j=m-1;j>=0;j--){
v[j].push_back(i);
i--;
}
tmp = i;
}
flag=0;
// pf(tmp);
res = (1+tmp)*tmp/2/m; //因为上面的循环中每次减小都是m的整数倍,而sum也是m的整数倍,所以两者之差也是m的整数倍
memset(f,0,sizeof f);
dfs(tmp);
printf("YES\n");
for(int i=0;i<m;i++){
int size = (int)v[i].size();
printf("%d",size);
for(int j=0;j<size;j++){
printf(" %d",v[i][j]);
}
printf("\n");
}
}
return 0;
}