0409小红书笔试

第三题红蓝球

Python模拟超时 只过了55%
有无大佬指点下怎么优化

//题目描述:小明有一个神奇的袋子和一堆编号为1到n的球。每个球都被涂成了红色或者蓝色,且这些球的表面上都分别写着一个数字。当红球在袋子里的时候,每过去一秒上面的数字就会增加1,蓝球上的数字则会减少1。小明会时不时向袋子里放入球或取出球,并且他想知道某些时刻袋子中球上写的数字之和。
//输入描述:第一行有一个正整数n(1<=n<=5000),代表小明有几个球。第二行有n个范围在[-1000000,1000000]内的整数,第i个代表编号为1到n的球上写的数。第三行是一个长度为n的仅由`R`和`B`构成的字符串,第i个字母代表编号为i的球是红色(R)或蓝色(B)第四行有一个正整数m(1<=m<=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
全部评论
那个样例我都看蒙圈了,第一次询问的时候,为啥我算出来是4 还有不是只询问3次吗,为啥输出有4个数
点赞 回复 分享
发布于 2023-04-09 20:18 北京
正常模拟就行
点赞 回复 分享
发布于 2023-04-09 18:22 北京
第三题都没看 直接提前退场了
点赞 回复 分享
发布于 2023-04-09 18:18 湖北

相关推荐

中途加入只写了1,3,第二题没看懂要干啥第一题:优先队列 + 贪心可以想到贪心,先把所有的数给拆出来,比如[11, 299],拆成[9,9,2,1,1],然后贪心,把大的数放在位数高的那个元素就行了,比如例子中可以变成[(1,2), (2,3)]表示第一个数有两位,第二个数有三位,然后放入优先队列,首先取出第二个元素,将第三位变成9,现在第二个元素只剩两位了,继续放入优先队列,按照该顺序写就OK。代码如下:#include <bits/stdc++.h>using namespace std;const int N = 2e5 + 10;int a[N] ,h[10];pair<int, int> b[N];int main() {int n;cin >> n;priority_queue<pair<int, int>> pq;for (int i = 0; i < n; i ++) {cin >> a[i];b[i].second = i;while (a[i]) {h[a[i] % 10] ++;a[i] /= 10;b[i].first ++;}pq.push(b[i]);}while(pq.size() > 0) {pair<int, int> t = pq.top();pq.pop();int num = 0;for (int i = 9; i >= 0; i --) {if (h[i] != 0) {h[i] --;num = i;break;}}a[t.second] = a[t.second] * 10 + num;t.first--;if (t.first !=0) {pq.push(t);}}long long res = 0;for (int i = 0; i < n; i ++) res += a[i];cout << res;}第三题:组合数首先要推公式,如果两个数a[i]和a[j]之间有n个0,设z = a[j] - a[i], 有多少种可能的序列,设为f(z, n),C是组合数。当n = 1 时,记为f(z, 1),很简单,该0可能是a[i] - a[j]之间的任何数,有z + 1 种可能,f(z, 1) = z + 1 = C(z+1, 1)。当n = 2 时,固定第一个0,如果第一个0为x,那么第二个0就和n = 1 时一样,此时枚举第一个0从[0-z],f(z, 2) = f(z,1) +f(z-1,1) + ...+f(1, 1) + f(0, 1) = 1 +... +z+1 = C(z+2, 2)。同理, n=3时,同理枚举第一个0。f(z, 3) = f(z, 2) +... f(0, 2) = C(z+2, 2) + C (z + 1, 2) + ... C(2, 2) = C(z+3, 3)可以看出来f(z, n) = C(z + n, n) = (z +n)!/(n! & z!);#include <bits/stdc++.h>using namespace std;#define int long longconst int p = 1e9 + 7;const int N = 6e5 + 10;int a[N];int mul[N], dmul[N];int qs(int a, int b) {int res = 1;while (b) {if (b&1) res = res * a % p;a = a * a % p;b >>= 1;}return res;}int C(int n, int m) {int cc = 1;for (int i = n - m + 1; i <= n; i ++) cc = cc * i %p;return cc * dmul[m] % p;}int lucas(int n, int m) {if (m == 0) return 0;int a = n % p, b = m % p;if (b > a) return 0;return C(a, b) * lucas(n/p, m /p) % p;}signed main() {int n; cin >> n;vector<int> v;int res = 1;mul[0] = dmul[0] = 1;for (int i = 1 ; i < N ; i ++) {mul[i] = mul[i - 1] * i % p;dmul[i] = qs(mul[i], p - 2);}for (int i = 1 ;i <= n; i ++) cin >> a[i];for (int i = 1 ; i < n; ) {if (a[i] == 0) {int j = i;while (a[j] == 0) j++;int d = a[j] - a[i - 1];int z = j - i;res = res * C(z + d, z) % p;i = j;} else i++;}cout << res << '\n';}
投递小红书等公司10个岗位
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
3
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务