CodeForces - 635D 线段树

CodeForces - 635D 线段树

题目链接
题目大意：n天，不修机器之前机器可以生产preNum个物品，修完之后可以生产repairNum个物品，修机器需要repairTime天，m个询问。
1 day num 表示在第day天，他想要生产num个。
2 day 表示在第day天开始修机器。
当2询问时，输出在修机器的基础上的生产个数。
样例

5 4 10 1 6
1 1 5
1 5 5
1 3 2
1 5 2
2 1
2 2
输出
7
1

思路：一开始想到了线段树，1操作就是常规的单点更新，然后发现2操作要更新一大堆点，会超时。
后来看了别人的博客，发现之后在树中维护两个量，一个是当前节点没有修机器要产多少个产品，另一个是当前节点修完机器要产多少个产品。在查询的时候，用没有修机器之前的产量加上之后的就ok了。
代码

#include <bits/stdc++.h>
using namespace std;

#define lson root << 1
#define rson root << 1 | 1
#define MID int mid = (l + r) / 2

const int maxn = 2e5 + 10;
int finishNum, preNum;

struct node {
   int preNum, repairedNum;
   node() {preNum = 0; repairedNum = 0;}
} tree[maxn << 2];

void Insert(int root, int l, int r, int pos, int v) {
   if(pos > r || pos < l) return ;
   if(l == r) {
      tree[root].preNum = min(v + tree[root].preNum, preNum);
      tree[root].repairedNum = min(v + tree[root].repairedNum, finishNum);
      return ;
   }

   MID;
   Insert(lson, l, mid, pos, v);
   Insert(rson, mid + 1, r, pos, v);
   tree[root].preNum = tree[lson].preNum + tree[rson].preNum;
   tree[root].repairedNum = tree[lson].repairedNum + tree[rson].repairedNum;
}

int cnt;
int query(int root, int l, int r, int ql, int qr, int flag) {
    if(ql > r || qr < l) return 0;
    if(ql <= l && r <= qr) {
        if(!flag) return tree[root].preNum;
        else return tree[root].repairedNum;
    }

    MID;
    return query(lson, l, mid, ql, qr, flag) + query(rson, mid + 1, r, ql, qr, flag);
}

int main()
{
    int n, repairTime, opNum;
    scanf("%d %d %d %d %d", &n, &repairTime, &finishNum, &preNum, &opNum);

    while(opNum--) {
        int op, day;
        scanf("%d", &op);
        if(op == 1) {
            int num;
            scanf("%d %d", &day, &num);
            Insert(1, 1, n, day, num);
        }
        else {
            scanf("%d", &day);
            int sum = 0;
            cnt = 0;
            sum += query(1, 1, n, 1, day - 1, 0);
            sum += query(1, 1, n, day + repairTime, n, 1);
            printf("%d\n", sum);
        }
    }
}