2018 牛客多校第五场 subseq （线段树+离散）

https://www.nowcoder.com/acm/contest/143/H

 

题意：

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

 

思路：

先用线段树预处理好以a[i]为开头的严格递增子序列有多少个，主要数据会爆long long，取1e18即可，然后再找到第k小的那一组，如果找不到输出-1.

 

代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=5e5+10;
const ll inf=1e18;
vector<int>v;
int a[maxn];
int ans[maxn];
ll sum[maxn*4];
ll dp[maxn];
int cnt;
void update(int l,int r,int k,ll val,int x)
{
    if(l==r)
    {
        sum[x]=min(inf,sum[x]+val);
        return ;
    }
    int mid=(l+r)/2;
    if(k<=mid)update(l,mid,k,val,x*2);
    else update(mid+1,r,k,val,x*2+1);
    sum[x]=min(inf,sum[x*2]+sum[x*2+1]);
}
ll query(int l,int r,int L,int R,int x)
{
    if(L>R)return 0;
    if(L<=l&&R>=r)
    {
        return sum[x];
    }
    int mid=(l+r)/2;
    ll ans=0;
    if(L<=mid)ans=min(ans+query(l,mid,L,R,x*2),inf);
    if(R>mid)ans=min(ans+query(mid+1,r,L,R,x*2+1),inf);
    return ans;
}
int main()
{
    int n;
    ll k;
    scanf("%d%lld",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end()), v.erase(unique(v.begin(),v.end()),v.end());
    cnt=v.size();
    for(int i=n;i>=1;i--)
    {
        int p=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
        dp[i]=query(1,cnt,p+1,cnt,1);
        dp[i]++;
        update(1,cnt,p,dp[i],1);
    }
    int z=0;
    for(int i=1;i<=n;i++)
    {
        if(z==0||(z>0&&a[i]>a[ans[z]]))
        {
            if(k>dp[i])k-=dp[i];
            else
            {
                k--;
                ans[++z]=i;
            }
            if(k==0)break;
        }
    }
    if(k||z==0)printf("-1\n");
    else
    {
        printf("%d\n",z);
        for(int i=1;i<=z;i++)
        {
            if(i>1)printf(" ");
            printf("%d",ans[i]);
        }
        printf("\n");
    }
    return 0;
}