洛谷P1118数字三角形

要吸氧

二项式+permutation

疯狂卡常结果没啥卵用…

反正思路挺显然的

就是搞个全排列,然后对应位置乘杨辉三角qwq

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define maxn 20
using namespace std ;
int n , m , sum ;
int vis[maxn] , a[maxn] ;
int C[maxn][maxn] , f[maxn] ;
void pre(int n ) {
    for(int i = 0 ; i <= n ; i ++) {
        C[i][0] = 1 ;
        C[i][1] = C[i][i-1] = i ;
    }
    for(int i = 1 ; i <= n ; i ++) {
        for(int j = 0 ; j <= i ; j ++) {
            C[i][j] = C[i-1][j-1] + C[i-1][j] ;
        }
    }
}
int S = 0 ;
int main () {
    scanf("%d%d",&n,&sum) ;
    pre(n) ;
    for(int i = 1 ; i <= n ; i ++) {
        a[i] = i ;
        S += a[i] * C[n-1][i-1] ;
        //cout << C[n-1][i-1] << "*" ;
    }
    if(S == sum) {
        for(int i = 1 ; i <= n ; i ++) {
            printf("%d ",a[i]) ;
        }
        puts("") ;
    }else 
    while(next_permutation(a+1,a+1+n)) {
        S = 0 ;
        for(int i = 1 ; i <= n ; i ++) {
            S += a[i] * C[n-1][i-1] ;
        }
        if(S == sum) {
            for(int i = 1 ; i <= n ; i ++) {
                printf("%d ",a[i]) ;
            }
            puts("") ;
            return 0 ;
        }
    }
    return 0 ;
}