2019ICPC南京网络赛F题 Greedy Sequence

题意：

给定一个1-n的排列a，记f(a[i])为a[i]在a[i-k,i+k]范围内的前驱(即小于a[i]的最大的数),则f可以理解为找a[i]的下一个数…(后面被吃了

数据范围：

n=1e5

链接：

https://nanti.jisuanke.com/t/41303

题解：

滑动窗口+set二分

#include<iostream>
#include<algorithm>
#include<set>
#include<numeric>
using namespace std;
int a[100010];
int nxt[100010];
int ans[100010];

int dfs(int x){
    if(nxt[x]==0)return 1;
    if(ans[x])return ans[x];
    return ans[x]=dfs(nxt[x])+1;
}

int main(){
    ios::sync_with_stdio(0);cin.tie(0);
    int t;cin>>t;
    while(t--){
        int n,k;cin>>n>>k;
        for(int i=1;i<=n;i++)cin>>a[i];
        for(int i=1;i<=n;i++)ans[i]=0;

        set<int,greater<int> > s;
        for(int i=1;i<=k;i++)s.insert(a[i]);
        for(int i=1;i<=n;i++){
            if(i+k<=n)s.insert(a[i+k]);
            if(i-k-1>=1)s.erase(a[i-k-1]);

            //for(auto it=s.begin();it!=s.end();it++)cout<<*it<<" ";cout<<endl;

            auto it=s.lower_bound(a[i]-1);
            if(it==s.end())nxt[a[i]]=0;
            else nxt[a[i]]=*it;
        }
        //for(int i=1;i<=n;i++)cout<<nxt[a[i]]<<" ";cout<<endl;
        for(int i=1;i<=n;i++)ans[i]=dfs(i);
        for(int i=1;i<n;i++)cout<<ans[i]<<" ";
        cout<<ans[n]<<endl;
    }
}