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照片，算是了卻一樁事吧。