poj3187 Backward Digit Sums---dfs暴力枚举

这题感觉我的方法真的好死。。其实也可以用next_permutation函数全排列一遍 很方便 写出来也漂亮

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#define inf 0x3f3f3f3f
using namespace std;

int a[15],vis[15],n,s,sum,p;

int dfs(int step)
{
    int i;
    if(step==n)
    {
        switch(n)
        {
            case 3:s=a[1]+2*a[2]+a[3];break;
            case 4:s=a[1]+3*a[2]+3*a[3]+a[4];break;
            case 5:s=a[1]+4*a[2]+6*a[3]+4*a[4]+a[5];break;
            case 6:s=a[1]+5*a[2]+10*a[3]+10*a[4]+5*a[5]+a[6];break;
            case 7:s=a[1]+6*a[2]+15*a[3]+20*a[4]+15*a[5]+6*a[6]+a[7];break;
            case 8:s=a[1]+7*a[2]+21*a[3]+35*a[4]+35*a[5]+21*a[6]+7*a[7]+a[8];break;
            case 9:s=a[1]+8*a[2]+28*a[3]+56*a[4]+70*a[5]+56*a[6]+28*a[7]+8*a[8]+a[9];break;
            case 10:s=a[1]+9*a[2]+36*a[3]+84*a[4]+126*a[5]+126*a[6]+84*a[7]+36*a[8]+9*a[9]+a[10];
        }
        if(s==sum) return 1;
        else return 0;
    }
    for(i=1;i<=n;i++)
    {
        if(!vis[i])
        {
            vis[i]=1;
            a[p++]=i;
            if(dfs(step+1)) return 1;
            a[--p]=0;
            vis[i]=0;
        }
    }
    return 0;
}

int main()
{
    int i;
    while(~scanf("%d%d",&n,&sum))
    {
        if(n==2) printf("1 2\n");
        else if(n==1) printf("1\n");
        else
        {
            memset(vis,0,sizeof vis);
            p=1;
            dfs(0);
            for(i=1;i<n;i++)
                printf("%d ",a[i]);
            printf("%d\n",a[n]);
        }
    }
    return 0;
}