2019 年杭电多校第４场 hdu6616（构造）

题意：就是给你一个ｎ,k,ｋ满足一个性质为ｎ的约数，现在给你一个ｎ一个ｋ问你是否能把这ｎ个数分成ｋ组，每组ｎ/k个，每组的权值相等。
解法：我们仔细观察就会发现这ｎ个数是一个等差数列我们简单的求一下和判断一下这个和是否能整除于ｋ即可，若不能整除与ｋ则输出no即可，之后ｎ为偶数的情况时候我们直接一个蛇形构造即可完成对其　的构造，当ｎ为奇数时我们就要构造出前３列后面的列数为偶数根据ｎ为偶数的情况来进行一样的构造即可，这题的难点就在于构造前３列，例如ｎ等于１５　ｋ等于３　我们可以将其构造前列，第一列我们直接从１开始到ｋ　第二列我们从(k+1)/2这个位置开始构造到ｋ这个位置之后在从第二列的第一个位置开始构造到k/2这个位置,第３列的话我们可以通过求出前３列的和再除上ｋ求出每一组的总和sum,之后用sum减去每一组前两个数的值即可。剩下的列数直接根据ｎ为偶数那样构造即可.
代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
vector<int>v[N];
int main()
{
  //  freopen("1.txt","r",stdin);
   // freopen("1.out","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        long long n,k; scanf("%lld%lld",&n,&k);
        for(int i=1;i<=k;i++) v[i].clear();
      //  int t=n/k; 
         long long sum=n*(n+1)/2;
        if(sum%k) {puts("no");continue;}
        else if(k==1)
        {
            for(int i=1;i<=n;i++)
             v[1].push_back(i);
        }else if(n&1) 
        {
            long long  num=1;
            long long sm=k*3*(k*3+1)/2;
            sm/=k; 
            for(int i=1;i<=k;i++)
                v[i].push_back(num++); 
            for(int i=(k+1)/2;i<=k;i++) v[i].push_back(num++);
            for(int i=1;i<(k+1)/2;i++)v[i].push_back(num++);
            for(int i=1;i<=k;i++)
            {
                int tt=sm-v[i][0]-v[i][1];
                v[i].push_back(tt);
                num++;
            }
            int tm=1;
             while(1)
            {
                if(num>=n)break;
                if(tm&1)
                {
                    for(int i=1;i<=k;i++)
                     v[i].push_back(num++);
                }else 
                {
                    for(int i=k;i>=1;i--)
                    v[i].push_back(num++);
                }
                tm++;
            }
        }else 
        { // puts("bug");
            long long  num=1;long long  tt=1;
             while(1)
            {
                if(num>=n)break;
                 if(tt&1)
              { for(int i=1;i<=k;i++)
                 v[i].push_back(num++); 
              }else 
                {
                    for(int i=k;i>=1;i--)
                    v[i].push_back(num++);
                }
                tt++;
            }
        }
        puts("yes");
       for(int i=1;i<=k;i++)
        { for(int j=0;j<n/k;j++)
           if(j!=n/k-1)
          cout<<v[i][j]<<" ";
         else cout<<v[i][j]<<endl;
        }
    }
}