- 题目地址
- 题目分析:
这里剪枝最关键的一步,是在于解决已经出现过的情况,因为位置的变化会再次出现。
经过分析可以知道,从可以取的最大元素开始,我们以后再取也只能取等于小于它的元素,不能取更大的元素了。(否则必然大于n值,就是说可以取的大的元素被取走了,不可能再取比被取的元素更大的了,一取就大于了n。)
这里我用了use来控制选取元素不能比前面大。
现在一想,题目的输出要求其实就是在暗示我们按照这样的方式来解决这个问题。
- 我的代码
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
int n, k, p, maxlen = 0, minVal, maxVal;
vector<int>val, temp, ans;
void dfs(int v, int len, int level, int use);
int main()
{
scanf("%d %d %d", &n, &k, &p);
for (int i = 1; pow(i, p) <= n; i++)
val.push_back(pow(i, p));
dfs(0, 0, 0, val.size() - 1);
if (maxlen == 0) printf("%s", "Impossible");
else
{
auto it = ans.begin();
printf("%d = %d%c%d", n, *it, '^', p);
for (it++; it != ans.end(); it++)
printf(" + %d%c%d", *it, '^', p);
}
return 0;
}
void dfs(int v, int len, int level, int use)
{
if (level < k)
for (int i = use; i >= 0; i--)
{
minVal = v + (k - level) * 1, //上界下界控制,优化了一半的时间500ms->250ms。
maxVal = v + (k - level)*val[val.size() - 1];
if (v + val[i] <= n && minVal <= n && maxVal >= n)
{
temp.push_back(i + 1);
dfs(v + val[i], len + i + 1, level + 1, i);
}
}
else if (v == n && len > maxlen && level == k)
{
ans = temp;
maxlen = len;
}
if (v)temp.pop_back();//因为我们是从没有开始的,要保证最后安全跳出去,故if
return;
}