subseq 牛客网多校

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

首先数据有毒 不止1e5

题目要求下标字典序第k小的LIS 既然是以下标决定字典序 那就维护一个dp[i]代表以a[i]为首项的所有LIS个数 这个可以提前线段树之类的搞出来

为求第k小 故i从最小的下标1开始遍历 如果dp[i]<k 说明当前这个下标太小了不合要求 k减去dp[i] 如果以a[i]为首的LIS数量 否则说明字典序第k小的LIS就是以a[i]开头的 然后k-- 这个自减是为了减去只有a[i]自己的LIS 因为它肯定是最小的 要是自减以后k=0 那代表正好凑够 退出循环即可 否则就找除了独一个a[i]的第(k-1)小以a[i]开头的LIS 个人认为这个k--有些不太好理解(拙)

至于dp[i]爆longlong的情况 取个1e18即可 因为题目要求的k不超1e18 多了也用不上

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=1e18;

struct node
{
    int l;
    int r;
    ll val;
};

node tree[4000010];
ll dp[1000010];
ll k;
int a[1000010],tmp[1000010],ans[1000010];
int n,len;

void pushup(int cur)
{
    tree[cur].val=min(N,tree[2*cur].val+tree[2*cur+1].val);
}

void build(int l,int r,int cur)
{
    int m;
    tree[cur].l=l;
    tree[cur].r=r;
    tree[cur].val=0;
    if(l==r) return;
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
}

ll query(int pl,int pr,int cur)
{
    ll res;
    if(pl<=tree[cur].l&&tree[cur].r<=pr)
    {
        return tree[cur].val;
    }
    res=0;
    if(pl<=tree[2*cur].r) res=min(N,res+query(pl,pr,2*cur));
    if(pr>=tree[2*cur+1].l) res=min(N,res+query(pl,pr,2*cur+1));
    return res;
}

void update(int tar,ll val,int cur)
{
    if(tree[cur].l==tree[cur].r)
    {
        tree[cur].val=min(N,tree[cur].val+val);
        return;
    }
    if(tar<=tree[2*cur].r) update(tar,val,2*cur);
    else update(tar,val,2*cur+1);
    pushup(cur);
}

int main()
{
    int i,p;
    scanf("%d%lld",&n,&k);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        tmp[i]=a[i];
    }
    sort(tmp+1,tmp+n+1);
    len=unique(tmp+1,tmp+n+1)-tmp-1;
    build(1,len,1);
    for(i=n;i>=1;i--)
    {
        p=lower_bound(tmp+1,tmp+len+1,a[i])-tmp;
        if(p+1<=len) dp[i]=query(p+1,len,1);
        dp[i]++;
        update(p,dp[i],1);
    }
    p=0;
    for(i=1;i<=n;i++)
    {
        if(p==0||(p>0&&a[i]>a[ans[p]]))
        {
            if(k>dp[i]) k-=dp[i];
            else
            {
                k--;
                ans[++p]=i;
            }
            if(k==0) break;
        }
    }
    if(p==0||k!=0) printf("-1\n");
    else
    {
        printf("%d\n",p);
        for(i=1;i<=p;i++)
        {
            printf("%d",ans[i]);
            if(i<p) printf(" ");
            else printf("\n");
        }
    }
    return 0;
}