AtCoder Beginner Contest 380 (A~E)題解

ZhangDT發表於2024-11-17

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';
		}
	}
}

相關文章