Codeforces 635D Factory Repairs (树状数组)

题意

有一个机器，修好的时候一天可以生产a个没修好的时候每天可以生产b个，初始时机器是坏的，修好需要k天，这k天不能生产。

给出一些操作：

1,di,ai 表示在第di天新增了ai个订单，这ai个订单只能在一天完成。

2,pi 表示选择在第pi天修复机器。

对每个2操作输出最后的生产数量。

思路

树状数组维护两个sum，一个是维修前即a，一个维修后即b的。

然后查询的时候输出维修前的suma和维修后的sumb就可以了。

代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <list>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define LL long long
#define lowbit(x) ((x)&(-x))
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1|1
#define MP(a, b) make_pair(a, b)
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
const int maxn = 1e5 + 10;
const double eps = 1e-8;
const double PI = acos(-1.0);
typedef pair<int, int> pii;

LL sum[2*maxn][2];
int now[2*maxn];

void add(int x, int val1, int val2)
{
while (x < (maxn << 1))
{
sum[x][0] += val1;
sum[x][1] += val2;
x += lowbit(x);
}
}

LL query(int t, int x)
{
LL res = 0;
while (x > 0)
{
res += sum[x][t];
x -= lowbit(x);
}
return res;
}

int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int n, a, b, k, q;
cin >> n >> k >> b >> a >> q;
while (q--)
{
int t, d, order;
cin >> t;
if (t == 1)
{
cin >> d >> order;
int pre = now[d];
now[d] += order;
int or0 = min(now[d], a) - min(pre, a);
int or1 = min(now[d], b) - min(pre, b);
add(d, or0, or1);
}
else
{
cin >> d;
LL ans = query(0, d - 1) + query(1, n) - query(1, d + k - 1);
cout << ans << endl;
}
}
return 0;
}