2019 杭电多校 1003 Divide the Stones —— 暴力

This way

题意：

给你n个石头，重量从1-n，现在要将其分成k堆使得每堆的重量一样。如果可以输出每堆石子的编号

题解：

如果$(n+1)\ast n/2\%k$不等于0，那么就无法分，其它情况都可以。
如果n/k是偶数，就蛇形向上放，如果它是奇数，那么我们考虑前3层将其变为全部相等。手模之后发现第一层从1-k依次放，第二层只需要从最中间向右依次放，再从1-中间放，最后第三层一个一个弥补。之后就按照偶数的情况考虑

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+5;
vector<int>ans[N];
ll sum[N];
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        ll n,k,cnt=0;
        scanf("%lld%lld",&n,&k);
        if(n%k||((n+1)*n/2)%k)
        {
            printf("no\n");
            continue;

        }
        for(int i=1;i<=k;i++)
            ans[i].clear();
        if((n/k)%2)
        {
            if(n==k)
            {
                if(n==1)
                    printf("yes\n1\n");
                else
                    printf("no\n");
                continue;
            }
            if(k%2)
            {
                for(int i=1;i<=k;i++)
                    ans[i].push_back(++cnt),sum[i]=cnt;
                for(int i=k/2+1;i<=k;i++)
                    ans[i].push_back(++cnt),sum[i]+=cnt;
                for(int i=1;i<=k/2;i++)
                    ans[i].push_back(++cnt),sum[i]+=cnt;
                ll mx=(1+k*3)*k*3/2/k;
                for(int i=1;i<=k;i++)
                    ans[i].push_back(mx-sum[i]);
                cnt=3*k;
                for(int i=4;i<=n/k;i+=2)
                {
                    for(int j=1;j<=k;j++)
                        ans[j].push_back(++cnt);
                    for(int j=k;j>=1;j--)
                        ans[j].push_back(++cnt);
                }
            }
            else
                printf("no\n");
        }
        else
        {
            for(int i=1;i<=n/k;i+=2)
            {
                for(int j=1;j<=k;j++)
                    ans[j].push_back(++cnt);
                for(int j=k;j>=1;j--)
                    ans[j].push_back(++cnt);
            }
        }
        printf("yes\n");
        for(int i=1;i<=k;i++)
            for(int j=0;j<ans[i].size();j++)
                printf("%d%c",ans[i][j]," \n"[j==(int)ans[i].size()-1]);
    }
    return 0;
}
/*
freopen("in.txt","r",stdin);
    freopen("1.out","w",stdout);
*/