POJ 3468 A Simple Problem with Integers
- 题意:区间修改,区间查询
- 思路:模板题
!!!!!!!!!!!!!!这道题忘记了在query里边pushdown懒标记,也没有开long long,不仅debug了很久,还wa了一发
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>
#define INF 0x3f3f3f3f
#define P(x) x>0 ? x : 0
#define MID l + r >> 1
#define lsn rt << 1
#define rsn rt << 1 | 1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
using namespace std;
typedef long long ll;
typedef vector<int>:: iterator VITer;
const int maxN=1e5+5;
ll N,Q;
ll num[maxN];
ll tree[maxN<<2],lazy[maxN<<2];
void pushup(ll rt) { tree[rt] = tree[lsn]+tree[rsn]; }
void build_tree(ll rt,ll l,ll r)
{
if(l==r) { tree[rt] = num[l]; return; }
ll mid=MID;
build_tree(Lson);
build_tree(Rson);
pushup(rt);
}
void pushdown(ll rt,ll l,ll r)
{
if(lazy[rt])
{
ll mid=MID;
tree[lsn] += lazy[rt]*(mid-l+1);
tree[rsn] += lazy[rt]*(r-mid);
lazy[lsn] += lazy[rt];
lazy[rsn] += lazy[rt];
lazy[rt]=0;
}
}
void update_range(ll rt,ll l,ll r,ll ql,ll qr,ll val)
{
if(ql<=l && qr>=r) { tree[rt]+=val*(r-l+1); lazy[rt]+=val; return; }
pushdown(rt,l,r);
ll mid=MID;
if(qr<=mid) update_range(QL,val);
else if(ql>mid) update_range(QR,val);
else { update_range(QL,val); update_range(QR,val); }
pushup(rt);
}
ll query(ll rt,ll l,ll r,ll ql,ll qr)
{
if(ql <= l && qr >= r) { return tree[rt]; }
pushdown(rt,l,r);
ll mid=MID;
if(qr <= mid) return query(QL);
else if(ql > mid) return query(QR);
else return query(QL)+query(QR);
}
void init()
{
memset(lazy, 0, sizeof(lazy));
}
int main()
{
while(~scanf("%lld%lld", &N, &Q))
{
for(ll i=1; i<=N ; i++)
scanf("%lld", &num[i]);
init();
build_tree(1,1,N);
while(Q--)
{
char s[10];
scanf("%s",s);
if(s[0]=='C')
{
ll a,b,c;
scanf("%lld%lld%lld",&a, &b, &c);
update_range(1,1,N,a,b,c);
}
else if(s[0]=='Q')
{
ll aa,bb;
scanf("%lld%lld",&aa,&bb);
printf("%lld\n",query(1,1,N,aa,bb));
}
}
}
return 0;
}