【HDOJ6315】Naive Operations（线段树，树状数组）

题意：

两个序列a和b，初始a[i]=0，b[i]给定且为一个1到n的排列，要求维护以下两种操作：
1.区间[L,R]内a[i]加1

2.询问[L,R]内a[i]/b[i]（下取整）之和

n，q<=1e5

思路：

 实际上a[i]/b[i]的线段树可以改为树状数组，因为只需要支持单点修改和前缀区间求和

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
#define fi first
#define se second 
#define MP make_pair
const int N=110000;
int ad[N<<2],mn[N<<2],T[N],b[N],n,m,cas;

void pushdown(int p)
{
    if(ad[p])
    {
        ad[p<<1]+=ad[p];
        mn[p<<1]+=ad[p];
        ad[p<<1|1]+=ad[p];
        mn[p<<1|1]+=ad[p];
        ad[p]=0;
    }
}

void pushup(int p)
{
    mn[p]=min(mn[p<<1],mn[p<<1|1]);
}

void build(int l,int r,int p)
{
  ad[p]=0;
  if(l==r)
  {
    mn[p]=b[l];
    return;
  }
  int mid=(l+r)>>1;
  build(l,mid,p<<1);
  build(mid+1,r,p<<1|1);
  pushup(p);
}

void modify(int x)
{
    for(int i=x;i<=n;i+=i&-i) T[i]++;
}

int ask(int x)
{
  int ans=0;
  for(int i=x;i;i-=i&-i) ans+=T[i];
  return ans;
}

void update(int l,int r,int x,int y,int p)
{
  if((x<=l)&&(r<=y)) 
  {
    ad[p]--;
    mn[p]--;
    return;
  }
  pushdown(p);
  int mid=(l+r)>>1;
  if(x<=mid) update(l,mid,x,y,p<<1);
  if(y>mid) update(mid+1,r,x,y,p<<1|1);
  pushup(p); 
}

void clear(int l,int r,int p)
{
    if(l==r)
    {
      if(!mn[p])
      {
        mn[p]=b[l];
        modify(l);
      }
      return;
    }
    pushdown(p);
    int mid=(l+r)>>1;
    if(!mn[p<<1]) clear(l,mid,p<<1);
    if(!mn[p<<1|1]) clear(mid+1,r,p<<1|1);
    pushup(p);
}

int main()
{
 
  while(scanf("%d%d",&n,&m)!=EOF)
  {
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    build(1,n,1);
    for(int i=1;i<=n;i++) T[i]=0;
    for(int i=1;i<=m;i++)
    {
      char ch[10];
      scanf("%s",ch);
      if(ch[0]=='a')
      {
        int l,r;
        scanf("%d%d",&l,&r);
        update(1,n,l,r,1);
        clear(1,n,1);
      }
       else
       {
             int l,r;
             scanf("%d%d",&l,&r);
          printf("%d
",ask(r)-ask(l-1)); 
       }
    }
   }
   return 0;
}