为了更好的阅读体检,可以查看我的算法学习网
在线评测链接:P1179
题目描述
塔子哥是一个爱好神秘的人,他有一个神奇的盒子,这个盒子有一个特殊的功能,里面的玩具会随着时间的流逝而改变自己表面上的数字。
盒子里面有很多编号为 1 1 1 到 n n n 的玩具,每个玩具都被涂成了红色或蓝色。红玩具每过一秒表面上的数字就会增加 1 1 1,而蓝玩具每过一秒表面上的数字就会减少 1 1 1。
这个盒子的功能让塔子哥着迷不已,他会时不时地放入玩具或者取出玩具,并且想知道某些时刻盒子中玩具上写的数字之和。(注意:从盒子中拿出玩具之后,玩具的数字将会保留)
输入描述
第一行有一个正整数 n n n ( 1 ≤ n ≤ 5000 1\le n\le 5000 1≤n≤5000 ),代表塔子哥有几个玩具。
第二行有 n n n 个范围在 [ − 1000000 , 1000000 ] [-1000000,1000000] [−1000000,1000000] 内的整数,第 i i i 个代表编号为 1 1 1 到 n n n 的玩具上写的数。
第三行是一个长度为
n
n
n 的仅由R
和B
构成的字符串,第
i
i
i 个 字母代表编号为
i
i
i 的玩具是红色(R
)或蓝色(B
)
第四行有一个正整数 m m m ( 1 ≤ m ≤ 100000 1\le m\le 100000 1≤m≤100000 ),代表塔子哥进行了几次操作。
第五行有 m m m 个递增的正整数,第 i i i 个代表塔子哥进行的第 i i i 次操作时间点。每个时间点塔子哥只会进行至多一次操作。时间点的范围在 [ 1 , 1000000000 ] [1,1000000000] [1,1000000000] 内。
第六行有 m m m 个整数,第 i i i 个代表塔子哥进行的操作。 0 0 0 为询问当前时间点袋中玩具上的数字之和,正数 x x x 代表放入了编号为x的玩具,负数 − x -x −x 代表取出了编号为 x x x 的玩具。
最开始盒子是空的。 你可以认为玩具上的数字变化均发生在时间点之前,而每次操作均发生在时间点之后。输入保证操作合法。
输出描述
设塔子哥进行了 k k k 次询问。你需要在一行中先输出 k k k ,然后输出 k k k 个数,第 i i i 个代表第 i i i 次询问的答案。题目保证塔子哥进行过至少一次询问。
样例
输入
3
-5 4 6
RBR
9
1 2 3 4 5 6 8 13 14
1 3 0 2 -3 0 -1 0 -2
输出
3 4 2 -5
题目思路
假设某时刻我们的盒子里有 x x x个红,他们的下标分别是 i 1 , i 2 , . . . , i x i_1,i_2,...,i_x i1,i2,...,ix,
y y y个蓝的,他们的下标分别是 j 1 , j 2 , . . . , j y j_1,j_2,...,j_y j1,j2,...,jy。
假设现在的时间是 t i ti ti ,
这些玩具进入箱子的时间是 i n _ t i m e in\_time in_time
那么答案就是一个公式:
∑ k = 1 x ( a i k + t i − i n _ t i m e i k ) + ∑ k = 1 y ( a j k − t i + i n _ t i m e j k ) \sum_{k=1}^{x}(a_{i_k} + ti - in\_time_{i_k}) + \sum_{k=1}^{y}(a_{j_k} - ti + in\_time_{j_k}) ∑k=1x(aik+ti−in_timeik)+∑k=1y(ajk−ti+in_timejk)
我们以红色的为例,将求和符号分配一下,如下:
∑ k = 1 x ( a i k + t i − i n _ t i m e i k ) \sum_{k=1}^{x}(a_{i_k} + ti - in\_time_{i_k}) ∑k=1x(aik+ti−in_timeik)
= ∑ k = 1 x a i k + ∑ k = 1 x t i − ∑ k = 1 x i n _ t i m e i k =\sum_{k=1}^{x}a_{i_k} + \sum_{k=1}^{x} ti - \sum_{k=1}^{x} in\_time_{i_k} =∑k=1xaik+∑k=1xti−∑k=1xin_timeik
不难发现,第一项是红色的价值总和, 第二项是 x ∗ t i x * ti x∗ti , 第三项是红色的进入盒子的时间的总和
不难发现,这三项都很容易在 O ( 1 ) O(1) O(1)的时间维护。(拿个变量存一下就行了啊) 。蓝色的是一样的道理,不再赘述。
具体细节看代码
代码
python代码
from collections import defaultdict
n = int(input())
a = list(map(int,input().split()))
s = input()
m = int(input())
t = list(map(int,input().split()))
op = list(map(int,input().split()))
# sr: 当前盒子中红色的价值总和
sr = 0
# nr: 当前盒子中红色的个数
nr = 0
# sir:当前盒子中红色的进入时间的总和
sir = 0
# 蓝色的三项
sb = 0
nb = 0
sib = 0
# 存一下当前某个玩具进入盒子的时间
in_time = defaultdict(int)
res = []
for i in range(m):
ti = t[i]
opt = op[i]
# 处理询问
if opt == 0:
# 公式如上
ans = nr * ti - sir + sr + sib - nb * ti + sb
res.append(ans)
# 处理操作
else:
# p 是 要放入 / 拿出 的玩具的下标
p = abs(opt) - 1
# 拿出
if opt < 0:
# 将玩具p拿出盒子,消除p的各项贡献
if s[p] == 'R':
sr -= a[p]
nr -= 1
sir -= in_time[p]
a[p] += ti - in_time[p]
in_time[p] = 0
else:
sb -= a[p]
nb -= 1
sib -= in_time[p]
a[p] -= ti - in_time[p]
in_time[p] = 0
else:
# 将玩具p加入到盒子中,更新各项贡献
if s[p] == 'R':
sr += a[p]
nr += 1
in_time[p] = ti
sir += in_time[p]
else:
sb += a[p]
nb += 1
in_time[p] = ti
sib += in_time[p]
res = [len(res)] + res
print(" ".join(map(str,res)))
C++代码 (from 2333)
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx,avx2,fma")
#include <bits/stdc++.h>
using namespace std;
#define out(x) cout << #x << '=' << x << endl
#define out2(x, y) cout << #x << '=' << x << ',' << #y << '=' << y << endl
#define no cout << "No" << endl; return
#define yes cout << "Yes" << endl; return
#define outvec(a) for (auto &v : a) { cout << v << ' '; } cout << endl
#define lowbit(x) (x & -x)
#define gcd __gcd
#define inf 0x3f3f3f3f3f3f3f3fLL
#define infi 0x3f3f3f3f
using ll = long long;
using pii = pair<int, int>;
void solve() {
int n;
cin >> n;
vector<int> num(n + 1);
for (int i = 1; i <= n; i++) {
cin >> num[i];
}
string color;
cin >> color;
color = " " + color;
int m;
cin >> m;
vector<int> time(m + 1), op(m + 1);
for (int i = 1; i <= m; i++) {
cin >> time[i];
}
for (int i = 1; i <= m; i++) {
cin >> op[i];
}
vector<int> has(n + 1);
ll sum = 0;
int pre = 0;
int red = 0;
int blue = 0;
vector<ll> ans;
for (int i = 1; i <= m; i++) {
sum += 1LL * red * (time[i] - pre);
sum -= 1LL * blue * (time[i] - pre);
if (op[i] == 0) {
ans.push_back(sum);
} else {
if (op[i] < 0) {
op[i] = -op[i];
int d = (time[i] - has[op[i]]);
sum -= num[op[i]];
if (color[op[i]] == 'R') {
sum -= d;
num[op[i]] += d;
red--;
} else {
sum += d;
num[op[i]] -= d;
blue--;
}
} else {
has[op[i]] = time[i];
sum += num[op[i]];
if (color[op[i]] == 'R') {
red++;
} else {
blue++;
}
}
}
pre = time[i];
}
cout << ans.size() << ' ';
outvec(ans);
}
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int t = 1;
//cin >> t;
while (t--) {
solve();
}
}