秋实大哥与小朋友 UESTC - 1059
Problem
秋实大哥以周济天下,锄强扶弱为己任,他常对天长叹:安得广厦千万间,大庇天下寒士俱欢颜。
所以今天他又在给一群小朋友发糖吃。
他让所有的小朋友排成一行,从左到右标号。在接下去的时间中,他有时会给一段区间的小朋友每人vv颗糖,有时会问第xx个小朋友手里有几颗糖。
这对于没上过学的孩子来说实在太困难了,所以你看不下去了,请你帮助小朋友回答所有的询问。
Input
第一行包含两个整数nn,mm,表示小朋友的个数,以及接下来你要处理的操作数。
接下来的mm行,每一行表示下面两种操作之一:
0 l r v : 表示秋实大哥给[l,r]这个区间内的小朋友每人v颗糖
1 x : 表示秋实大哥想知道第x个小朋友手里现在有几颗糖
1≤m,v≤1000001≤m,v≤100000,1≤l≤r≤n1≤l≤r≤n,1≤x≤n1≤x≤n,1≤n≤1000000001≤n≤100000000。
Output
对于每一个11 xx操作,输出一个整数,表示第xx个小朋友手里现在的糖果数目。
Sample Input
3 4
0 1 3 1
1 2
0 2 3 3
1 3
Sample Output
1
4
ps:树状数组
代码如下:
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>
#include<vector>
using namespace std;
const int maxn = 10000000+10;
map<int, int> mp;
vector<int> res;
struct node{
int l, r, v;
int x;
}node[100000+10];
int n, m;
int a[maxn];
inline int lowbit(int x){
return x&(-x);
}
inline void updata(int x,int v){
while(x <= n){
a[x] += v;
x += lowbit(x);
}
}
int inline sum(int x){
int res = 0;
while(x) {
res += a[x];
x -= lowbit(x);
}
return res;
}
int main() {
scanf("%d %d", &n, &m);
memset(a, 0, sizeof(a));
for(int j = 0; j < m; j++){
int op;
scanf("%d", &op);
if(op == 0){
int l, r, v;
scanf("%d %d %d", &l, &r, &v);
res.push_back(l);
res.push_back(r);
node[j].l = l;
node[j].r = r;
node[j].v = v;
}
else {
int x;
scanf("%d", &x);
res.push_back(x);
node[j].x = x;
}
}
sort(res.begin(), res.end());
for(int i = 0; i < res.size(); i++) {
mp[res[i]] = i + 1;
}
n = res.size();
for(int j = 0; j < m; j++) {
if(node[j].l == 0) {
printf("%d\n",sum(mp[node[j].x]));
}
else {
updata(mp[node[j].l], node[j].v);
updata(mp[node[j].r]+1, -node[j].v);
}
}
return 0;
}