PAT 1103. Integer Factorization (30)(dfs)

题意：给你n,k,p,让你求一个长度为k的数组a满足, n = a1^p + a2^p + .... ak^p, 如果有多个解，则取a的和最大的，如果还是有多解，则取字典序最大的。(n <= 400, k <= n, 1 < p <=7)

思路：因为数据量很小，所以可以爆搜，考虑有多解的情况需要选择和最大其次字典序最大，所以我们可以倒着搜。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 25;
const int maxm = 505;
int a[maxm], res[maxm], fac[maxn];
int n, p, k, tmpSum;
bool ok;

void dfs(int cur, int num, int sum, int val)
{
    if(sum > n) return ;
    if(num == k)
    {
        if(sum == n && val > tmpSum)
        {
            ok = 1;
            tmpSum = val;
            for(int i = 0; i < k; i++)
                res[i] = a[i];
        }
        else return ;
    }
    for(int i = cur; i >= 1; i--)
    {
        if(sum+fac[i] <= n)
        {
            a[num] = i;
            dfs(i, num+1, sum+fac[i], val+i);
        }
    }
}

int main(void)
{
    while(cin >> n >> k >> p)
    {
        for(int i = 1; i <= 20; i++)
        {
            int tmp = 1;
            for(int j = 1; j <= p; j++)
                tmp = tmp*i;
            fac[i] = tmp;
        }
        ok = tmpSum = 0;
        dfs(20, 0, 0, 0);
        if(ok)
        {
            printf("%d = ", n);
            for(int i = 0; i < k; i++)
            {
                printf("%d^%d", res[i], p);
                if(i != k-1) printf(" + ");
            }
            puts("");
        }
        else puts("Impossible");
    }
    return 0;
}