多校 hdu 6406 Taotao Picks Apples(线段树)
设d1[ i ]为从a1开始到 ai 的最长递增子序列长度,设d2[ i ]为从ai开始到an的最长递增子序列长度,假设要把ap改为q,可以把序列分为1到p-1和p+1到n两段,可以用线段树找到1到p-1最大值下标cur,然后ans+=d1[cur],如果q大于那个a[ cur ],ans还要+1,然后在区间p+1到n找到第一个大于max( q , a[cur] )的数的下标cur2,ans+=d2[ cur2 ]即可,可用线段树实现,可以倒过来遍历求d2数组,从线段树中找到第一个大于a[ i ]的数的下标cur,然后d2[ i ]=d2[ cur ]+1。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int a[maxn],d1[maxn],d2[maxn];
int tree[maxn*3],vis[maxn*3];
int ans,cur;
void query(int l,int r,int rt,int L,int R,int k)//求区间第一个大于k的数的下标
{
if(l==r) {cur=min(cur,l);return;}
int m=(l+r)>>1;
if(l>=L&&r<=R)
{
if(tree[rt<<1]>k) query(l,m,rt<<1,L,R,k);
else if(tree[rt<<1|1]>k) query(m+1,r,rt<<1|1,L,R,k);
return ;
}
if(L<=m&&tree[rt<<1]>k) query(l,m,rt<<1,L,R,k);
if(R>m&&tree[rt<<1|1]>k) query(m+1,r,rt<<1|1,L,R,k);
}
void query1(int l,int r,int rt,int L,int R)//求区间最大值的下标
{
if(l==r) {if(tree[rt]>a[cur]) cur=l;return;}
int m=(l+r)>>1;
if(l>=L&&r<=R)
{
if(tree[rt<<1]>=tree[rt<<1|1]) query1(l,m,rt<<1,L,R);
else query1(m+1,r,rt<<1|1,L,R);
return ;
}
if(L<=m) query1(l,m,rt<<1,L,R);
if(R>m) query1(m+1,r,rt<<1|1,L,R);
}
void build(int l,int r,int rt)
{
if(l==r) {tree[rt]=a[l];return ;}
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m,p,q,Max=0;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]>Max) d1[i]=d1[i-1]+1,Max=a[i];
else d1[i]=d1[i-1];
}
memset(tree,0,sizeof(tree));
build(1,n,1);
d2[n]=1;
for(int i=n-1;i;i--)
{
cur=n+1;
query(1,n,1,i+1,n,a[i]);//求区间第一个大于a[i]的数的下标
if(cur>n)
cur=0;
d2[i]=d2[cur]+1;
}
while(m--)
{
scanf("%d%d",&p,&q);
ans=cur=0;
if(p!=1)
query1(1,n,1,1,p-1);//求区间最大值的下标
ans+=d1[cur];
if(q>a[cur])ans++;
else q=a[cur];
cur=n+1;
if(p!=n)
query(1,n,1,p+1,n,q);//求区间第一个大于k的数的下标
if(cur<=n)
ans+=d2[cur];
printf("%d\n",ans);
}
}
}