bzoj 4345: [POI2016]Korale

最新推荐文章于 2018-10-27 16:52:00 发布

fyc_kabuto

最新推荐文章于 2018-10-27 16:52:00 发布

阅读量236

点赞数 1

分类专栏：线段树思路

本文链接：https://blog.csdn.net/qq_36808030/article/details/82695431

版权

思路同时被 2 个专栏收录

37 篇文章 0 订阅

订阅专栏

线段树

23 篇文章 0 订阅

订阅专栏

题意：

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

题解：

很久以前做过，完全忘了，于是有不会做……
先将那个值计算出来，考虑爆搜，状态就是 $（S,j）$ 表示当前乘积，最后一个数。
然后将结果排序……。于是用二元组表示 $(S,j)$ 表示同样的状态，放到堆里面，这个状态可以扩展出 $(S-a[j]+a[j+1],j+1),(S+a[j+1],j+1)$ ，扩展到第k个就可以了。
知道之后，那么就只用搜权值和 $1~k$ 的集合就可以了。用线段树优化即可。
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]);
}