西安邮电大学第五届ACM-ICPC校赛 D.最简单的一道题(线段树技巧)

题目连接
思路:一开始看到这题毫无思路,赛后看了题解和通过的代码,才了解到这个题的技巧。
显然修改操作可以通过线段树打标记实现。那么难的就是查询操作了。
考虑一种暴力的查询:每次询问的时候,我们把区间定在[x,n]上,然后类似dfs的一个过程在线段树上搜,每次搜到叶子节点的时候 ,我们判断[x,now]的最小值是不是等于now位置上的值。那这个复杂度显然是不行的。
那我们考虑怎样优化这个过程:
我们每次查询的时候都会递归左右子树,那有没有一种只需递归一边子树的方法呢?
显然是有的。
我们用一个数组g来记录一段区间[l,r]中的x数量,其中x满足:
1.l<=x<=r
2:t[x]=min(l~r)
显然叶子节点的g值为1,合并两个子树的时候显然左子树的答案是正确的,我们只需要另外计算一下右子树的贡献就行。
我们查询的时候,需要保证合法的查询区间的右端点是递增的,那么就要先递归左儿子再递归右儿子,然后我们维护一下前缀区间的最小值val,就可以计算这个区间的贡献了。
区间贡献计算方式:
显然如果当前点为叶子节点,那么如果这个点的值小于等于val的时候贡献为1,否则为0。
然后我们看这个区间的最小值如果不是小于等于val那么这个区间显然是不合法的,贡献必然为0。
反正我们看这个区间的左子树的最小值是不是小于等于val的,如果左子树的最小值小于等于val,那么我们只需要递归处理左子树,并加上右子树的贡献即可,因为此时右子树的g值必然是合法的,但是左子树的g值不一定合法;如果左子树的最小值不是小于等于val的我们就递归右子树即可,因为此时的左子树的贡献必然为0。
这个方法很巧妙的维护了一个g数组,来做到查询一个区间的贡献的时候只花费了log的时间,总的查询时间就是 l o g 2 log^2 log2.

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2e5 + 10;
#define fi first
#define se second
#define pb push_back
int n,m;
int a[N];
#define mid (l+r>>1)
#define ls o<<1
#define rs o<<1|1
int t[N<<2],laz[N<<2];
int g[N<<2];
void pd(int o,int l,int r){
  if(laz[o]){
    laz[ls]+=laz[o];
    laz[rs]+=laz[o];
    t[ls]+=laz[o];
    t[rs]+=laz[o];
    laz[o]=0;
  }
}
int merge(int o,int  l,int r,int x){
  //找到这个区间
  if(l==r)return t[o]<=x;
  pd(o,l,r);
  if(t[o]>x)return 0;
  if(t[ls]>x)return merge(rs,mid+1,r,x);
  return merge(ls,l,mid,x)+g[o]-g[ls];
}
void push_up(int o,int l,int r){
  t[o]=min(t[ls],t[rs]);
  g[o]=g[ls]+merge(rs,mid+1,r,t[ls]);
}
void build(int o,int l,int r){
  if(l==r){
    t[o]=a[l];
    g[o]=1;
    return ;
  }
  build(ls,l,mid);
  build(rs,mid+1,r);
  push_up(o,l,r);
}

void up(int o,int l,int r,int x,int y,int d){
  if(l>=x&&r<=y){
    laz[o]+=d;
    t[o]+=d;
    return ;
  }
  pd(o,l,r);
  if(x<=mid)up(ls,l,mid,x,y,d);
  if(y>mid)up(rs,mid+1,r,x,y,d);
  push_up(o,l,r);
}
int getx(int o,int l,int r,int x){
  if(l==r)return t[o];
  pd(o,l,r);
  int ans;
  if(x<=mid)ans=getx(ls,l,mid,x);
  else ans=getx(rs,mid+1,r,x);
  push_up(o,l,r);
  return ans;
}
int get(int o,int l,int r,int x,int y,int &val){
  if(l>=x&&r<=y){
    int ans=merge(o,l,r,val);
    val=min(val,t[o]);
    return ans;
  }
  pd(o,l,r);
  int ans=0;
  if(x<=mid)ans+=get(ls,l,mid,x,y,val);
  if(y>mid)ans+=get(rs,mid+1,r,x,y,val);
  push_up(o,l,r);
  return ans;
}
int main() {
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++)scanf("%d",a+i);
  build(1,1,n);
  for(int i=1;i<=m;i++){
    char x;
    scanf(" %c",&x);
    if(x=='c'){
      int l,r,k;
      scanf("%d%d%d",&l,&r,&k);
      up(1,1,n,l,r,k);
    }else{
      int x;
      scanf("%d",&x);
      int d=getx(1,1,n,x);
      if(x+1<=n)printf("%d\n",get(1,1,n,x+1,n,d));
      else puts("0");
    }
  }
  return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值