Backward Digit Sums(POJ-3187)

本来以为会超时,看来是想多了,63ms,看来对时间复杂度的判断能力还是不行啊。这个题正是next_permutation()函数的用武之地。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
int n,sum,a[100],A[100];
int main() {
    scanf("%d%d",&n,&sum);
    for(int i=0;i<n;i++) a[i] = i+1;
    do {
        for(int i=0;i<n;i++) A[i] = a[i];
        int cur = n;
        bool ok = false;
        if(n==1) { printf("1\n"); return 0; }
        while(true) {
            for(int i=0;i<cur-1;i++) A[i] = A[i]+A[i+1];
            if(cur==2) {
                if(A[0]==sum) ok = true;
                break;
            }
            cur--;
        }
        if(ok) {
            printf("%d",a[0]);
            for(int i=1;i<n;i++) printf(" %d",a[i]);
            printf("\n");
            break;
        }
    } while(next_permutation(a,a+n));
    return 0;
}


你可能感兴趣的:(ACM,poj)