DFS的题目有着很强的套路,在理解其递归套路的基础上可以解决很多DFS变形的题目。
题目描述:
题目大致意思:
用K个数的P次方来表示N,K个数中可以有相同的数(K个数按从大到小的顺序排列),有可能N有多种表示方法,这时输出K个数之和最大的,如果K个数之和相等,则输出前面的数字较大的。
大致思路:
这个是典型的DFS的问题,可以使用递归来解决。
提交结果如下:
有三个测试用例未通过,暂时放在第二轮刷题时解决。
提交的代码如下:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int n, k, p, max_sum = 0;
vector<int> temp, ans;
int arr[200];
void DFS(int index, int k_num, int sum, int temp_sum);
bool func(vector<int> arr1, vector<int> arr2);
bool cmp(int a, int b)
{
return a > b;
}
int main()
{
for (int i = 0;i < 200;i++)
{
arr[i] = i + 1;
}
cin >> n >> k >> p;
DFS(0, 0, 0, 0);
sort(ans.begin(), ans.end(), cmp);
cout<<n<<" = ";
for (int i = 0;i < ans.size()-1;i++)
{
cout<<ans[i]<<"^"<<p<<" + ";
}
cout<<ans[ans.size()-1]<<"^"<<p<<endl;
}
void DFS(int index, int k_num, int sum, int temp_sum)
{
if (k_num == k && sum == n)
{
if (temp_sum > max_sum)
{
ans=temp;
max_sum=temp_sum;
}
if(temp_sum==max_sum)
{
if (func(temp, ans))
{
ans = temp;
max_sum=temp_sum;
}
}
return;
}
if (sum > n || k_num > k||index ==(int)sqrt(n) )
return;
temp.push_back(arr[index]);//选第index个数,可以重复选
DFS(index, k_num + 1, sum + pow(arr[index], p), temp_sum + arr[index]);
temp.pop_back();//不选index个数
DFS(index + 1, k_num, sum, temp_sum);
}
bool func(vector<int> arr1, vector<int> arr2)
{
for (int i = arr1.size()-1;i >=0;i--)
{
if (arr1[i] > arr2[i])
return true;
else if (arr1[i] < arr2[i])
{
return false;
}
}
}
本次提交后,累计得分为1357。