AtCoder Beginner Contest 380
A - 123233
题意
给个\(6\)位数，判断是否是\(1\)个\(1\)，\(2\)个\(2\)，\(3\)个\(3\)。
思路
模拟。
代码
点击查看代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int, int> pii;
const int mxn = 1e6 + 5;
void solve()
{
string s;
cin >> s;
int a = 0, b = 0, c = 0;
for (int i = 0; i < s.length(); i++)
{
if (s[i] == '1')
{
a++;
}
else if (s[i] == '2')
{
b++;
}
else if (s[i] == '3')
{
c++;
}
}
if (a == 1 && b == 2 && c == 3)
{
cout << "Yes" << endl;
return;
}
cout << "No" << endl;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int T = 1;
//cin >> T;
while (T--)
{
solve();
}
return 0;
}
B - Hurdle Parsing header
题意
给个字符串，输出被"|"隔开的"-"每块有几个。
思路
模拟。
代码
点击查看代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int, int> pii;
const int mxn = 1e6 + 5;
void solve()
{
string s;
cin >> s;
int ans = 0;
for (int i = 1; i < s.length(); i++)
{
if (s[i] == '|')
{
cout << ans << " ";
ans = 0;
}
else
{
ans++;
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int T = 1;
//cin >> T;
while (T--)
{
solve();
}
return 0;
}
C - Move Segment
题意
给个\(01\)串，可以分为\(1\)-块和\(0\)-块(连续的\(1\)/\(0\))。现在将第\(k\)个\(1\)-块移动到第\(k-1\)个\(1\)-块之后，输出移动后的字符串。(题目保证至少有\(k\)个\(1\)-块)
思路
刚开始自己模拟\(wa\)了，后来发现题目已经告诉答案了：
那就记录一下每个\(1\)-块的起始、结束位置按上边的写。
代码
点击查看代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int, int> pii;
const int mxn = 1e6 + 5;
int st[mxn], ed[mxn];
void solve()
{
int n, k;
cin >> n >> k;
string s;
cin >> s;
int cnt = 1, t = 0;
for (int i = 0; i < s.length(); i++)
{
if (s[i] == '1')
{
if (!t)
{
st[cnt] = i;
}
t++;
}
else
{
if (t)
{
ed[cnt] = i - 1;
cnt++;
t = 0;
}
}
}
if (t)
{
ed[cnt] = s.size() - 1;
}
string ans = s;
for (int j = ed[k - 1] + (ed[k] - st[k]) + 2; j <= ed[k]; j++)
{
ans[j] = '0';
}
for (int i = ed[k - 1] + 1; i <= ed[k - 1] + (ed[k] - st[k]) + 1; i++)
{
ans[i] = '1';
}
cout << ans << endl;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int T = 1;
//cin >> T;
while (T--)
{
solve();
}
return 0;
}
D - Strange Mirroring
题意
给定字符串\(s\)，进行如下操作10^{100}次：颠倒\(s\)中的大小写得到\(t\)，令\(s = s + t\)。后有\(q\)次询问，输出每次询问\(s\)中第\(k\)个位置的字符。
思路
容易发现每次字符串的出都会\(×2\)，那就看要查询的位置是在当前字符串的前半段还是后半段，前者不需要颠倒大小写，后者需要，同时后半段对应前半段的位置为\(k - len\)。我们可以每次都让当前查询位置\(k\)落在后半段，每次都翻转一下。(开始写的循环有点问题，后面换成递归解决了)
代码
点击查看代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef pair<int, int> pii;
const int mxn = 1e6 + 5;
string s;
inline char change(char c)
{
return isupper(c) ? tolower(c) : toupper(c);
}
char f(int k, bool flag)
{
int len = s.size();
if (k < len)
{
return flag ? change(s[k]) : s[k];
}
while (k >= (len << 1))
{
len <<= 1;
}
return f(k - len, !flag);
}
void solve()
{
int q;
cin >> s >> q;
while (q--)
{
int k;
cin >> k;
k--;
putchar(f(k, false));
putchar(' ');
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int T = 1;
//cin >> T;
while (T--)
{
solve();
}
return 0;
}
E - 1D Bucket Tool
题意
给定\(n\)面墙，初始每面墙的颜色各不相同。进行\(q\)次操作，每次操作有两种形式：
\(1\ x\ c\) ：将第\(x\)和与其相邻的颜色相同的墙染成\(c\)这种颜色。
\(2\ c\) ：输出颜色为\(c\)的墙的数量。
思路
用并查集。记录每种颜色的根、大小(数量多少)、左右端点。详情见代码。
代码
点击查看代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mxn = 5e5 + 5;
int color[mxn], pre[mxn], siz[mxn], L[mxn], R[mxn], col_size[mxn];
int find(int x)
{
return x == pre[x] ? x : pre[x] = find(pre[x]);
}
void join(int a, int b)
{
int fa = find(a);
int fb = find(b);
if (fa != fb)
{
// 更新区间
L[fa] = min(L[fa], L[fb]);
R[fa] = max(R[fa], R[fb]);
siz[fa] += siz[fb];
pre[fb] = fa;
}
}
void paint(int x, int c)
{
int fx = find(x);
col_size[color[fx]] -= siz[fx]; // 原色块变小
color[fx] = c; // 染色
col_size[c] += siz[fx]; // 新色块变大
// 染色完检查两边颜色是否相同，相同则合并
int l = L[fx], r = R[fx];
if (color[find(l - 1)] == c)
{
join(l - 1, x);
}
if (color[find(r + 1)] == c)
{
join(r + 1, x);
}
}
void solve()
{
int n, q;
cin >> n >> q;
for (int i = 1; i <= n; i++)
{
pre[i] = color[i] = L[i] = R[i] = i;
siz[i] = col_size[i] = 1;
}
while (q--)
{
int cmd, x, c;
cin >> cmd;
if (cmd == 1)
{
cin >> x >> c;
paint(x, c);
}
else
{
cin >> c;
cout << col_size[c] << endl;
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int T = 1;
//cin >> T;
while (T--)
{
solve();
}
return 0;
}
