【HDOJ6616】Divide the Stones（构造）

题意：给定n堆石子，第i堆的个数为i，要求构造出一种方案将其分成k堆，使得这k堆每堆数量之和相等且堆数相等

保证k是n的一个约数

n<=1e5

思路：先把非法的情况判掉

n/k为偶数的方法及其简单

n/k为奇数就先构造出n=3k的情况然后减掉，剩下的就是n/k为偶数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<ll,ll> Pll;
typedef vector<int> VI;
typedef vector<PII> VII;
//typedef pair<ll,ll>P;
#define N  1000010
#define M  200010
#define fi first
#define se second
#define MP make_pair
#define pb push_back
#define pi acos(-1)
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
#define lowbit(x) x&(-x)
#define Rand (rand()*(1<<16)+rand())
#define id(x) ((x)<=B?(x):m-n/(x)+1)
#define ls p<<1
#define rs p<<1|1

const int MOD=1e9+7,inv2=(MOD+1)/2;
      double eps=1e-4;
      int INF=1e9;
      int inf=0x7fffffff;
      int dx[4]={-1,1,0,0};
      int dy[4]={0,0,-1,1};

VI c[N];

int read()
{
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

int main()
{
    int cas=read();
    while(cas--)
    {
        int n=read(),k=read();
        ll s=1ll*n*(n+1)/2;
        if(s%k>0||k==n&&k>1)
        {
            printf("no
");
            continue;
        }
        printf("yes
");
        if(n==1&&k==1)
        {
            printf("1
");
            continue;
        }
        int t=n/k;
        rep(i,1,k) c[i].clear();
        if(t>=3&&t%2==1)
        {
            int x=k/2,m=n;
            rep(i,1,k)
            {
                c[i].pb(m);
                m--;
            }
            rep(i,x+1,k)
            {
                c[i].pb(m);
                m--;
            }
            rep(i,1,x)
            {
                c[i].pb(m);
                m--;
            }
            int y=m;
            per(i,k,x+1)
            {
                c[i].pb(m);
                m-=2;
            }
            m=y-1;
            per(i,x,1)
            {
                c[i].pb(m);
                m-=2;
            }
            n-=3*k;
        }
        t=0;
        rep(i,1,n/2)
        {
            t++;
            if(t==k+1) t=1;
            c[t].pb(i);
        }
        t=0;
        per(i,n,n/2+1)
        {
            t++;
            if(t==k+1) t=1;
            c[t].pb(i);
        }

        rep(i,1,k)
        {
            rep(j,0,(int)c[i].size()-1) printf("%d ",c[i][j]);
            printf("
");
        }
    }
    return 0;
}