Greedy Sequence（主席树-区间小于每个数的最大值）

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 int tot=0,t,n,k;
 4 const int maxn=1e5+5;
 5 struct Node{
 6     int l,r,sum;
 7 }A[maxn*32];
 8 int arr[maxn],root[maxn];
 9 int ma[maxn],res[maxn];
10 
11 void updata(int left,int right,int &now,int pre,int value){
12     A[++tot]=A[pre],now=tot,A[now].sum++;
13     if(left==right) return;
14     int mid=left+right>>1;
15     if(mid>=value) updata(left,mid,A[now].l,A[pre].l,value);
16     else updata(mid+1,right,A[now].r,A[pre].r,value);
17 }
18 
19 int query(int left,int right,int x,int y,int value){   //返回的小于某个数的最大值
20     if(A[y].sum-A[x].sum==0) return -1;   //当前区间的个数为0则减枝;
21     if(left==right){      //递归到叶子节点
22         return left<value?left:-1;
23     }
24     int mid=left+right>>1;
25     if(value<=mid+1||A[A[y].r].sum-A[A[x].r].sum==0){  //查找的数在左边或者右边没有shu
26         return query(left,mid,A[x].l,A[y].l,value);
27     }
28     int t=query(mid+1,right,A[x].r,A[y].r,value);  //先查找右边
29     if(t==-1) query(left,mid,A[x].l,A[y].l,value);
30     else return t;
31 
32 }
33 
34 int main(){
35     ios::sync_with_stdio(false);
36     cin>>t;
37     while(t--){
38         memset(root,0,sizeof(root)),tot=0,memset(A,0,sizeof(A));
39         cin>>n>>k;
40         for(int i=1;i<=n;i++) cin>>arr[i],ma[arr[i]]=i;
41         for(int i=1;i<=n;i++) updata(1,n,root[i],root[i-1],arr[i]);
42         res[1]=1;
43         for(int i=2;i<=n;i++){
44             int L=max(1,ma[i]-k);
45             int R=min(n,ma[i]+k);
46             int flag=query(1,n,root[L-1],root[R],i);
47             if(flag==-1) res[i]=1;
48             else res[i]=res[flag]+1;   //递推的公式
49         }
50         for(int i=1;i<=n;i++){
51             printf("%d%c",res[i],i==n?‘\n‘:‘ ‘);
52         }
53     }
54     return 0;
55 }

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 int t,n,m;
 4 const int maxn=1e5+5;
 5 int sum[maxn<<2];
 6 int ma[maxn],res[maxn],arr[maxn];
 7 void updata(int rt) { sum[rt]=max(sum[rt<<1],sum[rt<<1|1]); }  //区间存的是最大值
 8 void build(int l,int r,int rt){
 9     sum[rt]=0;
10     if(l==r) {return;}
11     int mid=l+r>>1;
12     build(l,mid,rt<<1);
13     build(mid+1,r,rt<<1|1);
14     updata(rt);
15 }
16 void add(int l,int r,int rt,int pos,int value){
17     if(l==r) {sum[rt]=value;return;}
18     int mid=l+r>>1;
19     if(mid>=pos) add(l,mid,rt<<1,pos,value);
20     else add(mid+1,r,rt<<1|1,pos,value);
21     updata(rt);
22 }
23 int query(int l,int r,int rt,int L,int R,int value){
24     if(L<=l&&R>=r) return sum[rt];
25     int mid=l+r>>1;
26     int ans=0;
27     if(mid>=L) ans=max(ans,query(l,mid,rt<<1,L,R,value));
28     if(R>mid) ans=max(ans,query(mid+1,r,rt<<1|1,L,R,value));
29     return ans;
30 }
31 
32 int main(){
33     ios::sync_with_stdio(false);
34     cin>>t;
35     while(t--){
36         cin>>n>>m;
37         for(int i=1;i<=n;i++){
38             cin>>arr[i];
39             ma[arr[i]]=i;
40         }
41         build(1,n,1);
42         for(int i=1;i<=n;i++){
43             int L=max(1,ma[i]-m);
44             int R=min(n,ma[i]+m);
45             int x=query(1,n,1,L,R,i);   //小于这个数的最大值
46             add(1,n,1,ma[i],i);   //后添加必定保证是小于其的数
47             res[i]=res[x]+1;
48         }
49         for(int i=1;i<=n;i++)
50             printf("%d%c",res[i],i==n?‘\n‘:‘ ‘);
51     }
52     return 0;
53 }