A1103 Integer Factorization

A1103 Integer Factorization

题意描述

给定一个整数n 将其分解为 k个正整数的p次幂之和

输入输出格式

输入 n k p

数据规模

n <= 400 k <=n p >=2 and p <7

算法设计

这个题中没有给每个正整数的取值范围 但我们可以想到 如果单个数的p次已经大于了n 显然就没必要再增大了 而且 n 小于 400 p 大于2
每个数的取值集合 不过包含16个数
搜索的思路有两种 第一种是枚举k个数 每个数都有fac.size()种可能 显然 这颗搜索树是fac.size() 叉的 而且 对于 169
167 3 这张极端情况 显然 我们搜索树的深度可以达到167层 应该是必超时的
我们换一种思路 此题无非就是要我们在fac.size() 中选出k个数 当然 这k个数是可以重复的 那可以枚举每个数选或不选
来做搜索 这样每个节点就是二叉的 而且搜索树深度也不过是fac.size() 层 这样类似于完全背包的思路 是可以通过的
我们在DFS中添加两个参数，一个是index 表示我们已经选了几个数 一个是nowK 表示我们现在是在枚举哪个数
由题中要求 要求降序排列 那我们就逆序枚举fac 即可 而且 这样做 字典序一定是最大的

c++11代码

#include <bits/stdc++.h>
using namespace std;
using gg = long long;
gg n,k,p,rsum = -1;
vector<gg>fac,res,tmp;
void init(){
    gg i = 0,tmp = 0;
    while(tmp <= n){
        fac.push_back(tmp);
        tmp = pow(++i,p);
    }
}
void DFS(gg index,gg nowsum,gg fsum,gg nowk){
    if(index == k and nowsum == n){   //   先判断边界 优化解
        if(fsum > rsum){
            rsum =fsum;
            res = tmp;
        }
        return;
    }
    if(index >= k or nowsum >= n or nowk < 1) return;   //   剪枝
    tmp.push_back(nowk);
    DFS(index+1,nowsum + fac[nowk],fsum+nowk,nowk);   // 扩展 
    tmp.pop_back();
    DFS(index,nowsum,fsum,nowk-1);

}
int main()
{
   ios::sync_with_stdio(false);
   cin.tie(0);
   cin>>n>>k>>p;
   init();
   DFS(0,0,0,fac.size()-1);
   if(not res.size()) cout<<"Impossible";
   else{
       cout<<n<<" = ";
       for(gg i=0;i<res.size();i++){
           if(i) cout<<" + ";
           cout<<res[i]<<"^"<<p;
       }
   }
   return 0;
}

题目链接