PTA甲级考试真题练习103——1103 Integer Factorization

题目

思路

先记录p次方后不超过给定数字的最大数，然后进入DFS遍历，每次选择比当前容量小的最大数再往下层进行遍历，下层遍历条件为该层当前数字为顶点，该层容量减去该层p次方后的容量。
直到路径数组中恰好为k个数字，并且该层遍历到的数字恰好等于该层容量的时候存入最后的排序数组中。
注意减枝，进入DFS先判断path数组大小，如果已经大于给定k个数字则直接返回。

代码

#include <iostream>
#include <vector>
#include<algorithm>
using namespace std;
const int nmax = 10;
int num, n, p,maxn;
int sum = 0;
vector<int> path;
vector<pair<int, vector<int>>> vec;

void DFS(int cur,int cap) {
	if (path.size() >= n)
		return;
	for (int i = cur; i > 0; --i) {
		int ans = pow(i, p);
		if (ans == cap && path.size()==n-1) {
			path.emplace_back(i);
			sum += i;
			vec.emplace_back(pair<int, vector<int>>(sum, path));
			path.pop_back();
			sum -= i;
			return;
		}
		else if (ans < cap) {
			path.emplace_back(i);
			sum += i;
			DFS(i, cap - ans);
			path.pop_back(); 
			sum -= i;
		}
	}
}
bool cmp(const pair<int, vector<int>>& a, const pair<int, vector<int>>& b) {
	if (a.first != b.first)
		return a.first > b.first;
	else {
		for (int i = 0; i < a.second.size() && b.second.size(); ++i) {
			if (a.second[i] != b.second[i])
				return a.second[i] > b.second[i];
		}
	}
}
int main()
{
	cin >> num >> n >> p;
	maxn = pow(num,1.0/p);
	DFS(maxn,num);
	if (vec.empty()) {
		cout << "Impossible";
		return 0;
	}
	sort(vec.begin(), vec.end(), cmp);
	cout << num << " = " << vec[0].second[0] << "^" << p;
	for (int i = 1; i < vec[0].second.size(); ++i) {
		cout << " + " << vec[0].second[i] << "^" << p;
	}
	return 0;
}