AtCoder Beginner Contest 380 （A~E）题解
A - 123233
遍历字符串统计出现次数即可。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int n, m, k;
int a[N];
signed main() {
string s;
cin >> s;
map<char, int> mp;
for (auto t : s) {
mp[t]++;
}
if (mp['2'] == 2 && mp['1'] == 1 && mp['3'] == 3) {
cout << "Yes\n";
} else {
cout << "No\n";
}
}
B - Hurdle Parsing
直接计数，遇到 $ | $ 输出即可。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int n, m, k;
int a[N];
signed main() {
string s;
cin >> s;
int cnt = 0, f = 0;
for (int i = 0; i < s.size(); i++) {
if (s[i] == '|') {
if (f == 1)
cout << cnt << ' ';
cnt = 0;
f = 1;
}
if (s[i] != '|') {
cnt++;
}
}
}
C - Move Segment
题意是将第 $ k $ 个连续的字符 $ 1 $ 段与第 $ k-1 $个段合并。
用 $ vector $ 存储一下连续的每一段然后根据题目要求顺序即可。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int n, m, k;
int a[N];
signed main() {
string s;
cin >> n >> k;
cin >> s;
string now1 = "", now2 = "";
vector<string> a;
for (int i = 0; i < n; i++) {
if (s[i] == '1') {
now1 += s[i];
if (now2.size() != 0) {
// cout << now2 << '\n';
a.push_back(now2);
now2 = "";
}
}
if (s[i] == '0') {
now2 += s[i];
if (now1.size() != 0) {
a.push_back(now1);
// cout << now1 << '\n';
now1 = "";
}
}
}
if (now1.size())
a.push_back(now1);
if (now2.size())
a.push_back(now2);
// for (auto t : a) {
// cout << t << ' ';
// }
cout << '\n';
int cnt = 0, d1 = 0, d2 = 0;
for (int i = 0; i < a.size(); i++) {
if (a[i][0] == '1') {
cnt++;
if (cnt == k - 1) {
d1 = i;
}
if (cnt == k) {
d2 = i;
for (int j = 0; j <= d1; j++) {
cout << a[j];
}
cout << a[d2];
for (int j = d1 + 1; j < a.size(); j++) {
if (j != d2) {
cout << a[j];
}
}
return 0;
}
}
}
}
D - Strange Mirroring
利用对称的思想，观察每次查询的第 $ k $ 个字符最初是由哪个字符演变过来的，并记录演变次数，根据次数判断字符大小写。可以观察发现，每次我们都可以通过对称使得字符串长度减少 $ n*2^x $ 的长度，该长度通过二分查找，多次缩短我们便能找到查询位置字符最初的字符。
时间复杂度：$ O(q*log k) $。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int n, m, k;
int a[N];
int b[100];
signed main() {
string s;
cin >> s;
n = s.size();
s = ' ' + s;
int q;
cin >> q;
vector<int> a;
int x = n;
for (int i = 1; i <= 64; i++) {
if (x > 1e18) break;
a.push_back(x);
x *= 2;
}
for (int i = 0; i < a.size(); i++) {
b[i + 1] = a[i];
}
while (q--) {
cin >> k;
int d = 0;
int cnt = 0;
for (int i = 1; i <= 100; i++) {
int x = lower_bound(b + 1, b + 1 + (int)a.size(), k) - b;
x -= 1;
k -= b[x];
// cout << x << '\n';
if (b[x] != 0)
cnt++;
if (k <= n) break;
}
// cout << k << '\n';
// cout << cnt << '\n';
char x = s[k];
if (cnt % 2 == 0) {
cout << x << ' ';
} else {
if (x >= 'a' && x <= 'z')
cout << (char)(x - 32) << ' ';
else
cout << (char)(x + 32) << ' ';
}
// cout << '\n';
}
}
E - 1D Bucket Tool
使用两个 $ set $ 维护 $ l $ 和 $ r $ 区间，用 $ set<pair<int,int>> $ 维护存在的颜色的区间，使用 $ map<pair<int,int>> $ 维护颜色区间的颜色。每次修改通过二分查找出 $ x $ 分类讨论所在区间段再根据左右区间段颜色 $ c $ ，判断是否需要合并区间,每次修改完成后更新 $ sum $ 数组的值，不断维护 $ stl $ 容器的值，注意不要出现越界的问题，思路不乱维护好就可以了。
真是一次酣畅淋漓的 $ set $ 练习
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int n, m, k;
int sum[N];
set<pair<int, int>> col;
set<int> sel, ser;
map<pair<int, int>, int> mp;
void work(pair<int, int> l, pair<int, int> b, pair<int, int> r, int x) {
sel.erase(l.first), sel.erase(b.first), sel.erase(r.first);
ser.erase(l.second), ser.erase(b.second), ser.erase(r.second);
int a = mp[ {l.first, l.second}], bt = x, c = mp[ {r.first, r.second}];
int be = mp[ {b.first, b.second}];
sum[be] -= b.second - b.first + 1;
sum[x] += b.second - b.first + 1;
if (a == bt && bt == c) {
pair<int, int> f = {l.first, r.second};
mp.erase(b), mp.erase(l), mp.erase(r);
mp[f] = x;
col.erase(l), col.erase(r), col.erase(b);
col.insert(f);
sel.insert(l.first), ser.insert(r.second);
} else if (a == bt) {
pair<int, int> f = {l.first, b.second};
mp.erase(b), mp.erase(l);
col.erase(b), col.erase(l);
mp[f] = x;
col.insert(f);
sel.insert(l.first), ser.insert(b.second);
sel.insert(r.first), ser.insert(r.second);
} else if (bt == c) {
pair<int, int> f = {b.first, r.second};
mp.erase(b), mp.erase(r);
col.erase(b), col.erase(r);
mp[f] = x;
col.insert(f);
sel.insert(b.first), ser.insert(r.second);
sel.insert(l.first), ser.insert(l.second);
} else {
pair<int, int> f = {b.first, b.second};
mp[f] = x;
sel.insert(r.first), ser.insert(r.second);
sel.insert(l.first), ser.insert(l.second);
sel.insert(b.first), ser.insert(b.second);
}
}
signed main() {
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
cin >> n >> m;
for (int i = 1; i <= n; i++) {
sel.insert(i), ser.insert(i);
col.insert({i, i});
mp[ {i, i}] = i;
sum[i] += 1;
}
while (m--) {
int op, x, c;
cin >> op;
if (op == 1) {
cin >> x >> c;
auto it1 = sel.lower_bound(x);
auto it2 = ser.lower_bound(x);
if (it1 == sel.end()) {
it1 = --sel.end();
}
if (*it1 > x) it1--;
auto it3 = col.lower_bound({*it1, *it2});
pair<int, int> b = *it3, l, r;
pair<int, int> ed = *--col.end(), be = *col.begin();
if (b != ed) {
it3++;
auto t = col.lower_bound(*it3);
it3--;
r = *t;
}
if (b != be) {
it3--;
auto t = col.lower_bound(*it3);
it3++;
l = *t;
}
work(l, b, r, c);
} else {
cin >> c;
cout << sum[c] << '\n';
}
}
}
