Codeforeces Gym - 101234G Dreamoon and NightMarket [想法]

题意：给你一个数列，求所有组合的第k小。

题解：首先肯定是排序，由于k只有10^6，我们可以依次找最小。如何找最小呢？我们定义一个节点是数的总和与最大数的位置，每一个最小值能得到两种数，假设当前为（w,i），那它可以得到两个数，w+a[i+1]与w+a[i+1]-a[i]。然后通过优先队列依次取出最小值，直到第k个。（multiset会Tle）

AC代码：

#include<stdio.h>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node
{
	ll w,pos;
	node(){}
	node(ll w,ll pos)
	{
		this->w=w;
		this->pos=pos;
	}
};
inline char nc(){  
    static char buf[100000],*p1=buf,*p2=buf;  
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;  
}  
inline int _read(){  
    char ch=nc();int sum=0;  
    while(!(ch>='0'&&ch<='9'))ch=nc();  
    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();  
    return sum;  
}  

priority_queue<node>st;
bool operator<(node a,node b)
{
	return a.w>b.w;
}
ll a[200005];
int main()
{
	ll n,m;
	n=_read();
	m=_read();
	//scanf("%lld%lld",&n,&m);
	for(ll i=0;i<n;i++)a[i]=_read();
		//scanf("%lld",&a[i]);
	sort(a,a+n);
	st.push(node(a[0],0));
	for(ll i=0;i<m;i++)
	{
		if(i==m-1)
		{
			printf("%lld\n",st.top().w);
			return 0;
		}
		node k=st.top();
		st.pop();
		if(k.pos>=n-1)continue;
		st.push(node(k.w+a[k.pos+1],k.pos+1));
		st.push(node(k.w+a[k.pos+1]-a[k.pos],k.pos+1));
	}
}