题目链接:hdu4546
题目大意:给n长度的数字序列,从中取出k个元素(0<k<=n)求和,找出k个元素之和为第m小的和值。
解题思路:第一次做这样组合的题,第一直觉会想用爆搜穷举出所有组合,然后扔进multiset中维护这些值,搜索的复杂度是指数级,内存也超限。发现找到第m个小的可以直接用优先队列来模拟,因为m<10000,所以可以模拟到第m大就结束(其实用multiset也可以,不过没想出来)。思路是这样的:初始状态为0,每次出队的值一定是将来加上又一个值后最小的,然后将来的那个值要的情况和不要的情况分别入队,从而为下下次的选取提供基础比如:(1 2 3 4 5),1要入队(sum = 1),1不要入队(sum = 0);2要入队(sum = 2),2不要入队(sum = 0);3要入队(sum = 3),3不要入队(sum = 0);4要入队(sum = 4),4不要入队(sum = 0)…无形之间就把(1),(2),(3),(4),…,(1,2),(1,3)…都考虑完毕,这里有个前提:有序。(Ps:不要试图用前缀和枚举,因为中间间隔的组合方案没能考虑到,只能是连续的)
AC代码1
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <string>
#include <deque>
#include <queue>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <utility>
using namespace std;
const int maxn = (int)1e4+5;
int n,m; //n个题目 第m小
int num[maxn];
//int prefix_sum[maxn];
//multiset<int,less<int> > v;
struct node {
int now_point; //当前位置
int now_point_sum; //到达当前指针内容的总和
node(): now_point(0),now_point_sum(0) {
}
node(int now_point, int now_point_sum): now_point(now_point),now_point_sum(now_point_sum) {
}
~node() {
}
bool operator > (