蓝彗星（2022牛客寒假算法基础集训营4）

题目：蓝彗星

记录一下自己的错误历程(￣m￣）

第一次直接用开了一个vis数组写的，vis数组判断此时间点有蓝彗星无红彗星，像这样：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
typedef long long ll;
int vis[200010];
string s;
int n, t;
int start[100010];
int main()
{
	cin >> n >> t >> s;
	int max = 0;
	for (int i = 0; i < n; i++) { cin >> start[i]; if (max < start[i] && s[i] == 'B') max = start[i]; }//max表示有流星出现的最大时间
	for (int i = 0; i < n; i++) {//下面套一个循环直接遍历，判断
		if (s[i] == 'B') {
			for (int j = start[i]; j < start[i] + t; j++) 
				vis[j] = 1;
		}
	}
	for (int i = 0; i < n; i++) {
		if (s[i] == 'R') {
			for (int j = start[i]; j < start[i] + t; j++) 
				vis[j] = 0;
		}
	}
	int ans = 0;
	for (int i = 1; i < max + t; i++) if (vis[i]) ans++;
	cout << ans;
	return 0;
}

然后……然后当然是超时了＞﹏＜

于是乎，瞄了眼题解，改用差分，用了一个sum1和sum2数组，sum1数组表示蓝彗星,sum2数组表示红彗星

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
typedef long long ll;
int n, k;
string s;
int sum1[100010], sum2[100010];
int main()
{
	cin >> n >> k >> s;
	int x;
	int MAX = 0;
	for (int i = 0; i < n; i++) {
		cin >> x;
		MAX = (MAX < x) ? x : MAX;//MAX表示有彗星出现的最大时间
		if (s[i] == 'B') sum1[x]++, sum1[x + k]--;//差分处理
		else if (s[i] == 'R') sum2[x]++, sum2[x + k]--;
	}
	int ans = 0;
	for (int i = 1; i < MAX + k; i++) {
		sum1[i] += sum1[i - 1];
		sum2[i] += sum2[i - 1];
		if (sum1[i] && !sum2[i]) ans++;
	}
	cout << ans;
	return 0;
}

信心满满的交上去，WA,怎么肥事！

 

后来发现虽然数据范围是1e5，但数组中进行了x+k，所以数组要开2e5；

最终ac版本：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
typedef long long ll;
int n, k;
string s;
int sum1[200010], sum2[200010];
int main()
{
	cin >> n >> k >> s;
	int x;
	int MAX = 0;
	for (int i = 0; i < n; i++) {
		cin >> x;
		MAX = (MAX < x) ? x : MAX;
		if (s[i] == 'B') sum1[x]++, sum1[x + k]--;
		else if (s[i] == 'R') sum2[x]++, sum2[x + k]--;
	}
	int ans = 0;
	for (int i = 1; i < MAX + k; i++) {
		sum1[i] += sum1[i - 1];
		sum2[i] += sum2[i - 1];
		if (sum1[i] && !sum2[i]) ans++;
	}
	cout << ans;
	return 0;
}