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 long
using 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]);
}