hdu5355Cake 构造+dfs

//n块蛋糕，大小从1到n
//将其平均分为m块
//比赛的时候用贪心策略从大往小凑成这个蛋糕水过
//结果发现是spj出问题，这样的贪心策略有问题
//因为前面的蛋糕凑好可能会影响后面使得后面的凑不出来
//数据:
/* 17 23 6 27 7 28 7 31 8 32 8 35 9 36 9 39 10 40 10 45 9 53 9 59 10 71 9 99 9 99 10 100 10 100000 2 */
//正解是对于大于4*m的n可以将2*m将他们分为2*m的一组
//然后每一个在这2*m中取出最小和最大，次小和次大.....
//由于m小于等于10 , 最终n的长度在2*m~4*m之间
//直接爆搜即可
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std ;
const int maxn = 100010 ;
typedef long long ll ;
int a[20][maxn] ;
int vis[100] ;
int len[maxn] ;
ll ave ;ll sum ;
ll n , m ;
bool dfs(int sum , int pos , int num)
{
    if(num == m + 1)return true ;
    if(sum == ave)return dfs(0 , n , num + 1);
    for(int i = pos; i > 0;i--)
    {
        if(vis[i] || sum + i > ave)continue ;
        a[num][++len[num]] = i ;
        vis[i] = 1 ;
        if(dfs(sum+i , i - 1 , num))
        return true ;
        len[num]-- ;
        vis[i] = 0 ;
        if(i + sum == ave)return false;
        if(sum == 0)return false ;
    }
    return false ;
}
int main()
{
    //freopen("in.txt" , "r" , stdin) ;
    int t ;
    scanf("%d" , &t) ;
    while(t--)
    {
        memset(len , 0 , sizeof(len)) ;
        scanf("%lld%lld" , &n , &m) ;
        sum = n*(n+1)/2 ;
        ave = sum/m;
        if(ave * m != sum || n > ave)
        {
            puts("NO") ;
            continue ;
        }
        while(n > 4*m)
        {
            for(int j = 1;j <= m;j++)
            {
                a[j][++len[j]] = n - j + 1 ;
                a[j][++len[j]] = n - 2*m + j ;
            }
            ave -= (n + n - 2*m + 1);
            n -= 2*m ;
        }
        memset(vis ,0 , sizeof(vis)) ;
        dfs(0 , n ,1) ;
        puts("YES");
        sum = 0 ;
        for(int i = 1;i <= m;i++)
        {
            printf("%d" , len[i]) ;
            for(int j = 1;j <= len[i];j++)
            printf(" %d" , a[i][j]) ;
            puts("") ;
        }
    }
}