洛谷P1118 [USACO06FEB]数字三角形 搜索

洛谷P1118 [USACO06FEB]数字三角形Backward Digit Su…     搜索

这题我们发现每一个位置的加权就是 杨辉三角 yh[ n ][ i ]
然后我们就可以求 n! 暴力 ，但是会 TLE 额 好像是会T 因为12! 已经 4亿了
然后我们加一个强力剪枝 如果当前求出来的 s 已经大于 sum了，因为没有负的加
权，也就是说这一路是没有用了的，在继续搜下去也不能更新答案了，那么就直接退出 。

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <string>
#include <iostream>
#include <iomanip>
#include <algorithm>
using namespace std ;

int n,sum ;
int ans[13],yh[13][13] ;
bool finished ;
bool used[13] ;

inline void dfs(int p,int s)
{
    if(p>n) 
    {
        if(s==sum) 
        {
            finished = 1 ;
            for(int i=1;i<=n;i++) printf("%d ",ans[ i ]) ;
            exit(0) ;
        }
        return ;
    }
    for(int i=1;i<=n;i++) 
    {
        if(!used[ i ]) 
        {
            if(s+yh[n][p]*i>sum) continue ;     //剪枝  即不能更新答案了  
            ans[ p ] = i ; 
            used[ i ] = 1 ;
            dfs(p+1,s+yh[n][p]*i) ;
            used[ i ] = 0 ;
        }
    } 
}

int main() 
{
    scanf("%d%d",&n,&sum) ;
    yh[1][1] = 1 ;
    for(int i=2;i<=n;i++) 
        for(int j=1;j<=i;j++) 
            yh[ i ][ j ] = yh[ i-1 ][ j-1 ]+yh[ i-1 ][ j ] ;
    
    dfs(1,0) ;
    
    return 0 ;
}