HDU 5355 Cake

这是昨天的多校试题，当时此题的spj有问题，大部分人都是水过去。你会发现，今天你拿你比赛时AC的代码再提交，可能就过不了了。很不幸，我就是其中一员，经过一个下午的调试，再加上请教“大爷”解决了爆栈的问题后，终于过了。这里再次感谢“大爷”。

题意，给你n块蛋糕蛋糕大小从1到n（1<=n<=100000），要你分给m（2<=m<=10）个人，每个人得到的蛋糕总大小要一样，不能将蛋糕切开，问能否办到。能，输出“YES”，并给出一组方案，不能，则输出“NO”。

我的解法：dfs回溯，当然要剪枝，否则就会TLE。

附上AC代码：

#pragma comment(linker, "/STACK:102400000,102400000")//解决爆栈问题，并且交的时候还必须用C++交，不能用G++
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn=(1e+5)+5;
const int maxp=15;
long long sum;
long long n;
bool visit[maxn];
int cc;
int num[maxp][maxn];
int en[maxp];
long long num_sum[maxn];
long long res;
bool dfs(int deep,int k,long long s)
{
    if(s==sum)
    {
        num[cc][deep]=k;
        en[cc]=deep;
        return true;
    }
    if(s>sum)
    {
        return false;
    }
    int S;
    if(n<sum-s) 
        S=n;
    else 
        S=sum-s;
    if(k-1<S) 
        S=k-1;
    for(int i=S;i>=1;i--)
    {
        if(res+s<sum||s+num_sum[i]<sum)return false;//剪枝，res存的是剩余蛋糕总大小，num_sum[i]是1到i所有整数之和
        if(visit[i])continue;//给过的蛋糕不再给
        visit[i]=1;
        res-=i;
        if(dfs(deep+1,i,s+i))
        {
            num[cc][deep]=k;
            return true;
        }
        res+=i;
        visit[i]=0;
    }
    return false;
}

int main()
{
    int cnt;
    num_sum[0]=0;
    for(int i=1;i<maxn;i++)
    {
        num_sum[i]=num_sum[i-1]+i;//注意这里不能写(i+1)*i/2,因为这个式子在还没赋值给num_sum[i]时，一直是用int算的，而算的结果超过了int的范围，除非把i类型改为long long
    }
    scanf("%d",&cnt);
    while(cnt)
    {
        cnt--;
        int p;
        scanf("%d %d",&n,&p);
        sum=num_sum[n];
        if(sum%p||n<=p)
        {
            puts("NO");
            continue;
        }
        res=sum;
        sum=sum/p;
        cc=0;
        bool flag=true;
        memset(num,0,sizeof(num));
        memset(en,0,sizeof(en));
        memset(visit,0,sizeof(visit));
        for(int i=1;i<=p;i++)
        {
            cc++;
            if(!dfs(0,n+1,0))
            {
                puts("NO");
                flag=false;
                break;
            }
        }
        if(flag)
        {
            puts("YES");
            for(int i=1;i<=p;i++)
            {
                printf("%d",en[i]);
                for(int j=en[i];j>=1;j--)
                {
                    printf(" %d",num[i][j]);
                }
                printf("\n");
            }
        }

    }
}

贴上AC照片，算是了却一桩事吧。