P1118 [USACO06FEB]数字三角形`Backward Digit Su`… (dfs)

https://www.luogu.org/problemnew/show/P1118

看的出来是个dfs 本来打算直接从下到上一顿搜索 但是不会

看了题解才知道系数是个杨辉三角.......

这样就好写了 但是还是踩了一些坑

一开始只有70分 是因为我最后再把a[i]求和 再判断的 改了之后80分......

加了一个剪枝：

if(s>sums)

    return;

如果当前和大于sum就返回

成功AC

#include<bits/stdc++.h>

using namespace std;

int n,sums,a[],flag,yh[][],vis[],s;

void dfs(int t)  //t标记当前确定了几个数

{

    int i;

    if(s>sums)

    return;

    if((t==n+)&&s==sums)

    {

        //cout<<a[0];

        for(i=;i<=n;i++)

        cout<<a[i]<<' ';

        cout<<endl;

        exit();

    }

    //else

    {

        for(i=;i<=n;i++)

        {

            if(!vis[i])

            {

                vis[i]=;

                a[t]=i;

                s+=yh[n][t]*a[t];

                dfs(t+);

                s-=yh[n][t]*a[t];

                vis[i]=;

            }

        }

    }

}

int main()

{

    int i,j;

    cin>>n>>sums;

    for(i=;i<=n;i++)

    {

        yh[i][]=;

        yh[i][i]=;

    }

    for(i=;i<=n;i++)

    for(j=;j<i;j++)

    {

        yh[i][j]=yh[i-][j-]+yh[i-][j];

        //cout<<yh[i][j];

    }

    dfs();

}