2019年杭电多校第四场 Divide the Stones HDU - 6616 （思维+构造）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6616

题意：给你n个石子，第i个石子重i，给你一个k，k是n的因子，分成k组，使得每一组的石子重量之和相等，如果可以分输出Yes，否则No

思路：先特判

1、n=1  yes

2、n=k时，no

3、n为偶数时，如果n/k 为奇数，no ，为偶数 yes  可以两头两头的输出

4、n为奇数时，先把前3k个石子分成相等的k组，剩下的n-3k 可以两头两头的输出了

 

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5 + 10;
vector<int> ans[maxn];
int main()
{
    int T, n, k;
    scanf("%d", &T);
    while(T--) {
        scanf("%d %d", &n, &k);
		for(int i = 0; i < k; i++) ans[i].clear();
		if(n == 1) {
            printf("yes\n");
            printf("1\n");
            continue;
        } 
        else if(n == k) {
			printf("no\n");
			continue; 
		}  
        else if(n & 1) { 
            printf("yes\n");
            int tmp = n / k;
            int temp = k + 1;
            for(int i = 0; i < k; i++) ans[i].push_back(i+1);
            for(int i = k - 1; i >= 0; i -= 2) ans[i].push_back(temp), temp++;
            for(int i = k - 2; i >= 0; i -= 2) ans[i].push_back(temp), temp++;
            temp = (3 * k + 1) * 3 / 2;
            for(int i = 0; i < k; i++)  ans[i].push_back(temp - ans[i][0] - ans[i][1]);
            int res = 1;
            for(int i = 0; i < k; i++) { 
                for(int j = 0; j < (tmp - 3) / 2; j++) {
                    ans[i].push_back(3 * k + res);
                    ans[i].push_back(n - res + 1);
                    res++;
                } 
            }
            for(int i = 0; i < k; i++) {
                int len = ans[i].size();
                for(int j = 0; j < len; j++)
                    printf("%d%c",ans[i][j]," \n"[j == len - 1]);
            }
        }
        else {
            if((n / k) & 1) printf("no\n");
            else {
                printf("yes\n");
                int x = n / k / 2;
                for(int i = 0; i < k; i++) {
                    for(int j = 1; j <= x; j++) {
                        printf("%d ", i * x + j);
                    }
                    for(int j = 1; j <= x; j++) {
                        printf("%d%c", n - (i * x + j - 1)," \n"[j==x]);
                    }
                }
            }
        }
    }
    return 0;
}
//