BJ模拟 第k小和【贪心+优先队列】

题目描述：

从n个数中选若干（至少1）个数求和，求所有方案中第k小的和（和相同但取法不同的视为不同方案）。n<=100000,k<=500000

解题思路：

考虑从下到大选数。
假设当前最小状态为(sum,i)，表示和为sum，选的最大数为i，那么它会引出两个状态(sum-a[i]+a[i+1],i+1)或(sum+a[i+1],i+1)，用堆维护即可。
这样可以覆盖所有情况且都是从小到大拓展状态，所以是正确的。

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;
ll getint()
{
    ll i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=100005;
int n,m1,m2,w[N];
ll k,a[N*10],b[N*10];
struct node
{
    ll x;int id;
    node(ll _x=0,int _id=0):x(_x),id(_id){}
    inline friend bool operator < (const node &a,const node &b){return a.x>b.x;}
};
priority_queue<node>q;
void dfs1(int x,ll sum)
{
    if(x==n/2+1){a[++m1]=sum;return;}
    dfs1(x+1,sum),dfs1(x+1,sum+w[x]);
}
void dfs2(int x,ll sum)
{
    if(x==n+1){b[++m2]=sum;return;}
    dfs2(x+1,sum),dfs2(x+1,sum+w[x]);
}
bool check(ll sum)
{
    ll res=0;
    for(int i=1,j=m2;i<=m1;i++)
    {
        while(a[i]+b[j]>sum&&j)--j;
        res+=j;
    }
    return res-1>=k;
}
void solve1()
{
    dfs1(1,0),dfs2(n/2+1,0);
    sort(a+1,a+m1+1),sort(b+1,b+m2+1);
    ll l=0,r=a[m1]+b[m2],ans;
    while(l<=r)
    {
        ll mid=l+r>>1;
        check(mid)?(ans=mid,r=mid-1):l=mid+1;
    }
    cout<<ans<<'\n';
}
void solve2()
{
    sort(w+1,w+n+1);q.push(node(w[1],1));
    for(int i=1;i<k;i++)
    {
        node t=q.top();q.pop();
        if(t.id==n)continue;
        q.push(node(t.x-w[t.id]+w[t.id+1],t.id+1));
        q.push(node(t.x+w[t.id+1],t.id+1));
    }
    cout<<q.top().x<<'\n';
}
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),k=getint();
    for(int i=1;i<=n;i++)w[i]=getint();
    if(n<=35)solve1();
    else solve2();
    return 0;
}