AtCoder Beginner Contest 321
A - 321-like Checker (abc321 A)
题目大意
给定一个数,问从高位到低位,数字是不是递减的。
解题思路
可以以字符串读入,然后依次判断即可。
神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
string s;
cin >> s;
auto ok = [&]() {
for (int i = 0; i < s.size() - 1; ++i) {
if (s[i + 1] >= s[i])
return false;
}
return true;
};
if (ok())
cout << "Yes" << '\n';
else
cout << "No" << '\n';
return 0;
}
B - Cutoff (abc321 B)
题目大意
给定\(n-1\)个数字,问最后一个数字最少是多少,使得你的分数不小于 \(x\)。
数字在 \([0,100]\)之间。分数是去掉最低最高后的和。
解题思路
因为范围不大,直接枚举最后一个数,找最大最小,然后求分数判断即可。
神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n, x;
cin >> n >> x;
vector<int> a(n);
for (int i = 0; i < n - 1; ++i)
cin >> a[i];
int ans = 0;
auto ok = [&]() {
auto tmp = a;
sort(tmp.begin(), tmp.end());
int sum =
accumulate(tmp.begin(), tmp.end(), 0) - tmp.front() - tmp.back();
return sum >= x;
};
for (; ans <= 100; ++ans) {
a[n - 1] = ans;
if (ok())
break;
}
if (ans == 101)
ans = -1;
cout << ans << '\n';
return 0;
}
C - 321-like Searcher (abc321 C)
题目大意
给定\(k\),问\(321-like\)的第 \(k\)小的数是多少。
\(321-like\)就是第一题的定义,从高位到低位逐数字递减。
解题思路
最大的是\(9876543210\),可以猜测数不多,直接搜索全部数出来,然后排序看第\(k\) 个是多少即可。
神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int k;
cin >> k;
LL s = 0;
vector<LL> ans;
function<void(int)> dfs = [&](int x) {
s = s * 10 + x;
ans.push_back(s);
for (int i = x - 1; i >= 0; --i)
dfs(i);
s /= 10;
};
for (int i = 9; i >= 0; --i) {
dfs(i);
}
sort(ans.begin(), ans.end());
cout << ans[k] << '\n';
return 0;
}
D - Set Menu (abc321 D)
题目大意
给定两个数组\(a,b\)和一个数\(p\),各从中取一个数\(x,y\),得到贡献 \(\min(x+y, p)\) 。问所有取法的贡献和。
解题思路
对数组\(b\)排序,枚举数组 \(a\),那么小于 \(p-a\)的 \(b\)的贡献是 \(a+b\),其余的贡献是 \(p\),因此二分找到这个界限,分别计算这两类的贡献和即可,前者是个关于\(b\)的前缀和,后者就是一个数组\(b\)中 \(\geq p-a\)的数量的统计。时间复杂度是 \(O(n\log m)\)。
也可以对\(a,b\)都排序,此时\(a\)枚举从大到小, \(b\)枚举从小到大,双指针也行。
神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n, m;
LL p;
cin >> n >> m >> p;
vector<LL> a(n), b(m);
vector<LL> sumb(m);
for (auto& i : a)
cin >> i;
for (auto& i : b)
cin >> i;
sort(b.begin(), b.end());
partial_sum(b.begin(), b.end(), sumb.begin());
LL ans = 0;
for (auto& i : a) {
LL r = p - i;
auto cnt = lower_bound(b.begin(), b.end(), r) - b.begin();
ans += 1ll * i * cnt + (cnt ? sumb[cnt - 1] : 0) + 1ll * p * (m - cnt);
}
cout << ans << '\n';
return 0;
}
E - Complete Binary Tree (abc321 E)
题目大意
给定一棵完全二叉树,问与点\(x\)的距离为\(k\)的个数。
有 \(t\)组。
解题思路
考虑点\(5\)距离为 \(k\)的点都在哪。
首先在以该点的子树内,深度为 \(k\)的点有 \(2^k\)(起始为\(0\)), 对应的标号都可以求出来,这样可以判断在\(n\)内的标号有多少个。
然后考虑该点的父亲的另一个节点(点\(4\))的子树内,此时我们要求的深度是 \(2^{k-2}\)的点的个数。
然后考虑该点的父亲的父亲的另一个节点(点\(3\))的子树内,此时我们要求的深度是\(2^{k-3}\)的点的个数。
不断往父亲考虑,因为每次往父亲走,点标号都会除以 \(2\),最多 \(\log\)次就考虑完了。
总的复杂度是 \(O(t\log n)\)
神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t;
cin >> t;
auto solve = [&](LL x, LL k, LL n) {
if (k < 0)
return 0ll;
if (k >= 64)
return 0ll;
__int128 num = (1ll << k), l = x;
l = (l << k);
__int128 r = l + num - 1;
if (l > n)
return 0ll;
LL ret = min(r, (__int128)n) - l + 1;
return ret;
};
while (t--) {
LL n, x, k;
cin >> n >> x >> k;
LL ans = 0;
if (k)
ans += solve(x, k, n);
while (x > 1 && k > 0) {
--k;
ans += solve((x ^ 1), k - 1, n);
x /= 2;
}
if (x && k == 0)
ans++;
cout << ans << '\n';
}
return 0;
}
F - #(subset sum = K) with Add and Erase (abc321 F)
题目大意
箱子,放球,拿球,球上有数,各不相同。
每次操作后,问里面球数和为\(k\)的方案数。
解题思路
不考虑拿走球的操作,每次放球,求方案数,就是一个经典的\(0/1\)背包。
设 \(dp[i]\)表示和为 \(j\)的方案数,考虑选不选这个球(其数字为\(x\)),则有 \(dp[i] = dp[i] + dp[i - x]\)
现在拿走这个球,得从中剔除选了这个球的方案数。
设\(dp2[i]\)表示不选该球,和为 \(i\)的方案数。由于 \(dp[i]\)的方案包括两部分:选了该球和没选该球的。只要减去选了该球的方案数,那就是没选该球的方案数。而选了该球的方案数,其实就是 \(dp2[i-x]\),即选了这个球后,剩下的 \(i-x\)就是由不选该球的方案数组成,这恰好是我们定义的 \(dp2[i-x]\)的定义。
于是 \(dp2[i] = dp[i] - dp2[i - x]\)转移式即可。
时间复杂度是 \(O(qn)\)。
这其实是最经典的退背包问题。
神奇的代码
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
const int mo = 998244353;
int main(void) {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int q, k;
cin >> q >> k;
vector<int> dp(k + 1, 0);
dp[0] = 1;
while (q--) {
string op;
int x;
cin >> op >> x;
if (op[0] == '+') {
for (int i = k; i >= x; --i) {
dp[i] += dp[i - x];
if (dp[i] >= mo)
dp[i] -= mo;
}
} else {
vector<int> dp2(k + 1, 0);
for (int i = 0; i <= k; ++i) {
if (i < x) {
dp2[i] = dp[i];
} else {
dp2[i] = dp[i] - dp2[i - x];
if (dp2[i] < 0)
dp2[i] += mo;
}
}
dp.swap(dp2);
}
cout << dp[k] << '\n';
}
return 0;
}
G - Electric Circuit (abc321 G)
题目大意
给定\(n\)个部件和 \(m\)条线。部件有接口,有红蓝两色,数量都是\(m\)。一条线链接的两个接口必须不同颜色,但可以同个部件。
将部件视为点,线视为边,随机连线,问联通块的期望个数。
解题思路
<++>
神奇的代码