题目大意:从N个正数的序列中,选出一些数,使它们的和等于给定值M,要求选出的一组数是所有可能的组合中最小的。所谓组合 A < 组合B,是指当A和B都是升序排列时,满足A[0] = B[0], A[1] = B[1]......A[k-1]=B[k-1], A[k] < B[k]......。
时间限制200ms,只要求输出最小的组合,直接的想法就是先按升序排序,然后从第一个元素开始dfs,直到搜索到和为M的组合,肯定是最小的组合。但这样做会有三个测试点超时,因此必须考虑剪枝。
① 如果之前已经找到解,那么直接return。设置一个引用变量 flag实现这个功能。
② 如果当前的和大于M,那么直接return。(这个我没有想到)
③ 此外,递归的边界 index == v.size(),需要写在判断当前和是否等于M之后,否则如果组合含有最后一个元素的话,会不经过判断直接return,从而丢解。
还有一个坑点是,最后一个测试点是所有数的和小于M的情形,这种情况不可能有解,但是用dfs会超时。因此在输入时就需要记录所有元素的和。
AC代码:
#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
#include<algorithm>
#include<cmath>
#pragma warning(disable : 4996)
using namespace std;
vector<int> coins;
void dfs(int index, int target, vector<int>&tmp, vector<int>&solution, bool& flag)
{
if (target == 0)
{
solution = tmp;
flag = true;
return;
}
if (target < 0 || flag || index == coins.size()) return;
tmp.push_back(coins[index]);
dfs(index + 1, target - coins[index], tmp, solution, flag);
tmp.pop_back();
dfs(index + 1, target, tmp, solution, flag);
}
int main()
{
int N, M;
scanf("%d%d", &N, &M);
coins.resize(N);
int sum = 0;
for (int i = 0; i < N; i++)
{
scanf("%d", &coins[i]);
sum += coins[i];
}
if (sum < M)
{
printf("No Solution");
return 0;
}
sort(coins.begin(), coins.end());
vector<int> tmp, solution;
bool flag = false;
dfs(0, M, tmp, solution, flag);
if (solution.empty())
{
printf("No Solution");
return 0;
}
for (int i = 0; i < solution.size(); i++)
{
printf("%d", solution[i]);
if (i < solution.size() - 1) printf(" ");
}
system("pause");
return 0;
}
也可以用动态规划解决。待补。