【备战秋招】每日一题:2023.04.09-春招-小红书笔试题(第三题)-神奇的盒子

最新推荐文章于 2024-05-19 21:34:53 发布

塔子哥学算法

最新推荐文章于 2024-05-19 21:34:53 发布

阅读量92

点赞数

文章标签： java 算法数据结构 c语言 c++ visual studio

本文链接：https://blog.csdn.net/m0_73659489/article/details/132218049

版权

为了更好的阅读体检，可以查看我的算法学习网
 在线评测链接:P1179

题目描述

塔子哥是一个爱好神秘的人，他有一个神奇的盒子，这个盒子有一个特殊的功能，里面的玩具会随着时间的流逝而改变自己表面上的数字。

盒子里面有很多编号为 $1$ 到 $n$ 的玩具，每个玩具都被涂成了红色或蓝色。红玩具每过一秒表面上的数字就会增加 $1$ ，而蓝玩具每过一秒表面上的数字就会减少 $1$ 。

这个盒子的功能让塔子哥着迷不已，他会时不时地放入玩具或者取出玩具，并且想知道某些时刻盒子中玩具上写的数字之和。(注意:从盒子中拿出玩具之后，玩具的数字将会保留)

输入描述

第一行有一个正整数 $n$ ( $1\le n\le 5000$ )，代表塔子哥有几个玩具。

第二行有 $n$ 个范围在 $[- 1000000, 1000000]$ 内的整数，第 $i$ 个代表编号为 $1$ 到 $n$ 的玩具上写的数。

第三行是一个长度为 $n$ 的仅由R和B构成的字符串，第 $i$ 个字母代表编号为 $i$ 的玩具是红色(R)或蓝色(B)

第四行有一个正整数 $m$ ( $1\le m\le 100000$ )，代表塔子哥进行了几次操作。

第五行有 $m$ 个递增的正整数，第 $i$ 个代表塔子哥进行的第 $i$ 次操作时间点。每个时间点塔子哥只会进行至多一次操作。时间点的范围在 $[1, 1000000000]$ 内。

第六行有 $m$ 个整数，第 $i$ 个代表塔子哥进行的操作。 $0$ 为询问当前时间点袋中玩具上的数字之和，正数 $x$ 代表放入了编号为x的玩具，负数 $- x$ 代表取出了编号为 $x$ 的玩具。

最开始盒子是空的。你可以认为玩具上的数字变化均发生在时间点之前，而每次操作均发生在时间点之后。输入保证操作合法。

输出描述

设塔子哥进行了 $k$ 次询问。你需要在一行中先输出 $k$ ，然后输出 $k$ 个数，第 $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$ 个红,他们的下标分别是 $i_1,i_2,...,i_x$ ,

$y$ 个蓝的,他们的下标分别是 $j_1,j_2,...,j_y$ 。

假设现在的时间是 $t i$ ,

这些玩具进入箱子的时间是 $in\_time$

那么答案就是一个公式:

$\sum_{k=1}^{x}(a_{i_k} + ti - in\_time_{i_k}) + \sum_{k=1}^{y}(a_{j_k} - ti + in\_time_{j_k})$

我们以红色的为例，将求和符号分配一下，如下：

$\sum_{k=1}^{x}(a_{i_k} + ti - in\_time_{i_k})$

$=\sum_{k=1}^{x}a_{i_k} + \sum_{k=1}^{x} ti - \sum_{k=1}^{x} in\_time_{i_k}$

不难发现，第一项是红色的价值总和，第二项是 $x * t i$ , 第三项是红色的进入盒子的时间的总和

不难发现，这三项都很容易在 $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();
	}
}