bzoj 4345: [POI2016]Korale

题意：

有n个带标号的珠子，第i个珠子的价值为a[i]。现在你可以选择若干个珠子组成项链（也可以一个都不选），项链的价值为所有珠子的价值和。现在给所有可能的项链排序，先按权值从小到大排序，对于权值相同的，根据所用珠子集合的标号的字典序从小到大排序。请输出第k小的项链的价值，以及所用的珠子集合。

题解：

挺好的题。
用堆维护二元组（x，a），表示当前合为x，最后一个数是num[a]（排序后）。那么每次找出最小的x，再加入（x+num[a+1],a+1）和（x+num[a+1]-num[a],a+1）。
正确性显然。
然后知道答案构造，可以dfs大爆搜，枚举所有可能的填法（线段树优化），找够了就退出。
code：

#include<queue>#include<cstdio>#include<cstdlib>#include<cstring>#include<iostream>#include<algorithm>#define LL long longusing namespace std;struct node{    LL a,x;    node(LL _a,LL _x):a(_a),x(_x){}};bool operator < (node a,node b) {return a.a>b.a;}priority_queue<node> q;LL n,k;LL a[1000010],num[1000010];LL ans[1000010],cnt=0;struct trnode{    LL lc,rc,c;}tr[2000010];LL tot=0;LL bt(LL l,LL r){    LL x=++tot;    if(l!=r)    {        LL mid=(l+r)/2;        tr[x].lc=bt(l,mid);        tr[x].rc=bt(mid+1,r);        tr[x].c=min(tr[tr[x].lc].c,tr[tr[x].rc].c);    }    else tr[x].c=a[l];    return x;}LL findans(LL x,LL l,LL r,LL k,LL c){    if(tr[x].c>c) return 0;    if(l==r) return l;    LL mid=(l+r)/2;    if(k<=mid)    {        LL t=findans(tr[x].lc,l,mid,k,c);        if(t) return t;    }    return findans(tr[x].rc,mid+1,r,k,c);}LL p[1000010],len=0;void dfs(LL x,LL num){    if(!cnt) return;    if(num==0)    {        cnt--;        if(!cnt)        {            for(LL i=1;i<=len;i++) printf("%lld ",p[i]);            exit(0);        }    }    LL i;    for(i=x+1;i<=n;i++)    {        i=findans(1,1,n,i,num);        if(i)        {            p[++len]=i;            dfs(i,num-a[i]);            len--;        }        else break;    }}int main(){    scanf("%lld %lld",&n,&k);k--;    for(LL i=1;i<=n;i++) scanf("%lld",&a[i]),num[i]=a[i];    sort(num+1,num+n+1);    q.push(node(num[1],1));    for(LL i=1;i<=k;i++)    {        node t=q.top();q.pop();        ans[i]=t.a;        if(i!=k&&t.x<n)        {            q.push(node(t.a+num[t.x+1],t.x+1));            q.push(node(t.a+num[t.x+1]-num[t.x],t.x+1));        }    }    for (LL i=k;i&&ans[i]==ans[k];i--) cnt++;    printf("%lld\n",ans[k]);    bt(1,n);dfs(0,ans[k]);}