PAT A 1103 Integer Factorization (30分)

一、题目概述

给定正整数N、K、P， 将N划分为K个数的和，且每个数均为某数的P次幂。

二、思路

使用DFS探测所有可能的组合。

递归探测，每次找到一个合适的值，加入数组temp，统计因数的和。当数组temp拥有K个P次幂相加为N的元素，且因数和>Max(当前最大因数和)时，保存temp为备选结果。

void dfs(int margin, int ceil, int sum)
{
    if(a.size() < K)
    {
        for(int i = ceil; i; --i)
            if(val[i] <= margin)
            {
                a.push_back(i);
                dfs(margin - val[i], i, sum + i);
                a.pop_back();
            }
    }
    else if(!margin && sum > max_sum)
    {
        ans = a;
        max_sum = sum;
    }
}

注意：测试点超时dfs中增加和大于N时返回的剪枝判断

三、代码

#include <cstdio>
#include <vector>
#include <cmath>
using namespace std;
int N, K, P, max_sum = 0;
vector<int> ans, a, val;
void dfs(int margin, int ceil, int sum)
{
    if(a.size() < K)
    {
        for(int i = ceil; i; --i)
            if(val[i] <= margin)
            {
                a.push_back(i);
                dfs(margin - val[i], i, sum + i);
                a.pop_back();
            }
    }
    else if(!margin && sum > max_sum)
    {
        ans = a;
        max_sum = sum;
    }
}
int main()
{
    scanf("%d %d %d", &N, &K, &P);
    for(int i = 0; (int)pow((double)i, P) <= N; ++i)
        val.push_back((int)pow((double)i, P));
    dfs(N, val.size() - 1, 0);
    if(!ans.size())
    {
        printf("Impossible");
        return 0;
    }
    printf("%d = ", N);
    for(int i = 0; i < ans.size(); ++i)
        printf("%d^%d%s", ans[i], P, i + 1 == ans.size() ? "":" + ");
}