牛客网暑期ACM多校训练营(第五场)H subseq(树状数组)

题意

给定一个序列 a[1..n],求下标字典序第 k 小的严格递增子序列

题解

考虑逐位确定,每次计算 a[i…n] 中,以a[i]这个数字为开头的严格递增子序列的个数,用树状数组统计,然后1…n与k比较,小于k就减去dp[i],否则就放a[i],当然要保证a[i]大于前一个放的数;这个树状数组和以前的不太一样,是记录大于等于x的数量,与一般的是倒过来的,虽然有些意外不过想了想确实是,以前的是记在区间的右端点,现在是记录在左端点

代码

#include<bits/stdc++.h>
#define N 1000005
#define P pair<int,int>
using namespace std;
typedef long long ll;
const int M=1e9+7;
const int inf=1e9+7;
ll c[N],dp[N];
int b[N],ans[N],a[N];
void add(int x,ll k,int n)
{
    while(x){
        c[x]+=k;
        if(c[x]>1e18)c[x]=1e18;
        x-=(x&-x);
    }
}
ll sum(int x,int n)
{
    ll ans=0;
    while(x<=n){
        ans+=c[x];
        x+=(x&-x);
        if(ans>1e18)ans=1e18;
    }
    return ans;
}
int main()
{
    int n;
    ll k;
    scanf("%d%lld",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    sort(b+1,b+n+1);
    int tot=unique(b+1,b+n+1)-b;
    for(int i=n;i;i--){
        a[i]=lower_bound(b+1,b+tot,a[i])-b;
        ll tmp=sum(a[i]+1,tot);
        dp[i]=tmp+1;
        add(a[i],tmp+1,tot);
    }
    int p=0;
    for(int i=1;i<=n&&k;i++){
        if(p&&a[i]<=a[ans[p]])continue;
        if(k>dp[i])k-=dp[i];
        else ans[++p]=i,k--;
    }
    if(!p||k)printf("-1\n");
    else {
        printf("%d\n%d",p,ans[1]);
        for(int i=2;i<=p;i++)
            printf(" %d",ans[i]);
        puts("");
    }
    return 0;
}

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值