秋实大哥与小朋友 UESTC - 1059

深藏阁楼爱情的钟 2022-06-14 10:39 119阅读 0赞

### **秋实大哥与小朋友 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;
    }