【题目链接】
http://acm.hdu.edu.cn/showproblem.php?pid=6315
题目意思
给定长度n的数列b[i],m次操作,1:给a[i]一个区间内都加上1;2:求和一个区间a[i]/b[i]向下取整。
解题思路
区间更新和区间查询,不难想到用线段树。现在就是怎么处理a[i]/b[i]。让线段树维护一个b[i]的最大值,一个a[i]的最小值,sum=a[i]/b[i]。当最大值max大于最小值min是就说明有个sum要加一了。只要查询到叶子节点更新下a[i]值和sum的值。(注意这里是把min加上a[i],不能用max减a[i],虽然值一样但是后者操作相当于让懒惰标记没有了,以为你大区间的max依旧不变,所以每次更新都要跑到底导致超时)
代码部分
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <string>
#include <map>
#include <math.h>
using namespace std;
#define LL long long
#define inf 0x3f3f3f3
#define lson l , m , rt<<1
#define rson m+1 , r , rt<<1|1
#define lrt rt<<1
#define rrt rt<<1|1
const int mod = 1e9+7;
const int N = 1e5+7;
int sum[N<<2];
int col[N<<2]; ///懒惰标记
int tmin[N<<2],tmax[N<<2];
int b[N];
void pushup (int rt) ///求和
{
sum[rt] = sum[lrt]+sum[rrt];
tmin[rt] = min(tmin[lrt],tmin[rrt]);
tmax[rt] = max(tmax[lrt],tmax[rrt]);
}
void build(int l,int r,int rt) ///建树
{
col[rt] = 0;
if (l == r)
{
sum[rt] = tmax[rt] = 0;
tmin[rt] = b[l];
return;
}
int m = (r+l)>>1;
build (lson);
build (rson);
pushup(rt);
}
void pushdown(int rt) ///下移懒人
{
if (col[rt])
{
tmax[lrt] += col[rt];
tmax[rrt] += col[rt];
col[lrt] += col[rt];
col[rrt] += col[rt];
col[rt] = 0;
}
}
void qudate(int L,int R,int l,int r,int rt)///区间跟新
{
if (L <= l && r <= R)
{
tmax[rt] ++;
if (tmax[rt] < tmin[rt])
{
col[rt] ++;
return;
}
if (l == r && tmax[rt] >= tmin[rt])
{
sum[rt] ++;
//tmax[rt] -= b[l]; ///超时
tmin[rt]+= b[l];
return;
}
}
pushdown(rt);
int m = (l+r)>>1;
if (L <= m) qudate(L,R,lson);
if (R > m) qudate(L,R,rson);
pushup(rt);
}
int query(int L,int R,int l,int r,int rt) ///区间求和
{
if (L <= l && r <= R)
return sum[rt];
pushdown(rt);
int m = (l+r)>>1;
int ans = 0;
if (L <= m) ans += query(L,R,lson);
if (R > m) ans += query(L,R,rson);
return ans;
}
int main()
{
int n,m,l,r;
char str[20];
while (~scanf("%d %d",&n,&m))
{
for (int i = 1; i <= n; i++)
scanf("%d",&b[i]);
build(1,n,1);
for(int i = 0; i < m; i++)
{
scanf("%s%d%d",str,&l,&r);
if (str[0] == 'a')
qudate(l,r,1,n,1);
else printf("%d\n",query(l,r,1,n,1));
}
}
return 0;
}