文章目录
Codeforces Round #678 (Div. 2)
A. Reorder
题意: 给定n和m,问 ∑ i = 1 n ∑ j = i n a j j \sum_{i=1}^{n}{\sum_{j=i}^{n}{\frac{a_j}{j}}} ∑i=1n∑j=injaj是否为m
题解: 由于 ∑ i = 1 n ∑ j = i n a j j \sum_{i=1}^{n}{\sum_{j=i}^{n}{\frac{a_j}{j}}} ∑i=1n∑j=injaj式子可以等于 a 1 1 ∗ 1 + a 2 2 ∗ 2 + . . . + a n n ∗ n = a 1 + a 2 + . . . + a n \frac{a1}{1}*1 + \frac{a2}{2}*2 + ... + \frac{an}{n}*n = a1+a2+...+an 1a1∗1+2a2∗2+...+nan∗n=a1+a2+...+an。因此只需要判断累加和是否为m即可
代码:
#include <bits/stdc++.h>
using namespace std;
int const N = 2e5 + 10;
typedef long long LL;
int n, T, m;
int main() {
// freopen("in.txt", "r", stdin);
cin >> T;
while(T--) {
cin >> n >> m;
LL sum = 0;
for (int i = 1; i <= n; ++i) {
int x;
cin >> x;
sum += x;
}
if (sum == m) cout << "YES\n";
else cout << "NO\n";
}
return 0;
}
B. Prime Square
题意: 给定n,问是否能够构造出一个方阵,使得方阵中任意行和任意列的和为素数,同时方阵内每个元素均不是素数
题解: 构造的时候只需要保证左斜对角线都为素数,其他元素都为1即可满足条件,因此左斜对角线填的元素可以是大于n的素数-n-1.因此,枚举素数,找到第一个符合条件的即可。这种构造题目好像都是这样考虑,每次一般都是1、2、3、4、素数、2进制构造。
代码:
#include <bits/stdc++.h>
using namespace std;
int const N = 2e5 + 10;
typedef long long LL;
int n, T, m;
int prime[N], cnt, st[N];
unordered_map<int, int> p;
int is_prime(int x) {
if (x == 2) return 1;
for (int i = 2; i <= sqrt(x); ++i) {
if (x % i == 0) return 0;
}
return 1;
}
void get_prime(int n) {
for (int i = 2; i <= n; ++i) {
if (!st[i]) prime[cnt++] = i, p[i] = 1; // 如果这个数字没有被记录,那么这个数字必然为素数,记录一下
for (int j = 0; prime[j] <= n/i; ++j) {
st[prime[j] * i] = true; // 筛掉pj*i这个合数
if (i % prime[j] == 0) break; // i%pj==0,说明pj是i的最小素因子,因此i*素数的最小素因子也是pj,在i递增的时候也会被筛掉,因此不需要在这里判断
}
}
}
int main() {
get_prime(N - 1);
// freopen("in.txt", "r", stdin);
cin >> T;
while(T--) {
cin >> n;
if (is_prime(n)) {
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= n; ++j) {
cout << 1 << " ";
}
cout << endl;
}
}
else {
int t = 0;
// cout << "p:" << t << endl;
// 1 1 1 2
// 1 1 2 1
// 1 2 1 1
// 2 1 1 1
for (int i = 0; i < cnt; ++i) {
int this_p = prime[i];
if (this_p <= n) continue;
if (!p.count(this_p - (n - 1))) {
t = this_p - (n - 1);
break;
}
}
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= n; ++j) {
if (j == n + 1 - i) cout << t << " ";
else cout << 1 << " ";
}
cout << endl;
}
}
}
return 0;
}
C.Binary Search
题意: 给定n,x,pos,现在给定一个数列,如果按照二分查找,必定能够在pos的位置找到x,问这样的全排列数列有多少个。
题解: 由于二分查找有着固定的方式,当前元素大于x,那么就去当前元素左边找;如果当前元素小于x,那么就去当前元素右边找。因此从一开始到达pos位置的路径是固定的,同时路径上面的点与x的大小关系也是固定的。所有可以先使用二分方法寻找到这条路径上面大于x的点的数目和小于x的点数目。那么所有数组的数目为 C A a a ! C B b b ! ( n − a − b − 1 ) ! C_{A}^{a} a! C_{B}^{b}b!(n-a-b-1)! CAaa!CBbb!(n−a−b−1)!,其中A为1~n中大于x的数目,a为二分查找时需要大于x的数目,B为1 ~ n中小于x的数目,b为二分查找时需要小于x的数目。
代码:
#include <bits/stdc++.h>
using namespace std;
int const N = 2e5 + 10, MOD = 1e9 + 7;
typedef long long LL;
int main() {
// freopen("in.txt", "r", stdin);
int n, x, pos;
cin >> n >> x >> pos;
int l = 0, r = n;
int A = x - 1, B = n - x;
int a = 0, b = 0;
while(l < r) {
int mid = (l +r) >> 1;
if (mid < pos) {
a ++;
l = mid + 1;
}
else if (mid > pos) {
b ++;
r = mid;
}
else if (mid == pos) {
l = mid + 1;
}
}
// cout << a << " " << b << endl;
LL sum = 1;
if (a) for (int i = A - a + 1; i <= A; ++i) sum = (sum * i) % MOD;
if (b) for (int i = B - b + 1; i <= B; ++i) sum = (sum * i) % MOD;
for (int i = 1; i <= n - a - b - 1; ++i) sum = (sum * i) % MOD;
cout << sum << endl;
return 0;
}
D. Bandit in a City
题意: 有一颗有根树,根为1,每个节点上面都有一个权值,这些权值都会不断向下面的分支移动,问使得每个叶节点上权值最大的值最小是多少。
题解: 可以统计每个子树的权值和是多少,那么每个叶节点最后的权值就是总的权值和/叶节点数目。这个结论一开始就是这么猜的,然后一直觉得证明不了,感觉无法让左边分支的节点走到右边分支,但是当这种情况出现时,答案会在下面的子树取得。证明不了,就口胡数学归纳法过了吧。
代码:
#include <bits/stdc++.h>
using namespace std;
int const N = 2e5 + 10;
typedef long long LL;
int n, T, m;
int e[N], ne[N], h[N], idx, v[N], sz[N];
LL sum[N];
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
LL ans = 0;
void dfs(int u) {
sum[u] = v[u];
if (h[u] == -1) sz[u] = 1;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
dfs(j);
sum[u] += sum[j];
sz[u] += sz[j];
}
LL tmp = sum[u] / sz[u] + (sum[u] % sz[u] != 0);
ans = max(ans, tmp);
}
int main() {
// freopen("in.txt", "r", stdin);
memset(h, -1, sizeof h);
cin >> n;
for (int i = 2; i <= n; ++i) {
int t;
scanf("%d", &t);
add(t, i);
}
for (int i = 1; i <= n; ++i) scanf("%d", &v[i]);
dfs(1);
cout << ans;
return 0;
}
E. Complicated Computations
题意: 给定一个数组a,求出数组所有子串的mex的mex值。 1 < = n < = 1 0 5 , 1 < = a i < = n 1<=n<=10^5, 1<=a_i<=n 1<=n<=105,1<=ai<=n
题解: 暴力的想法是求出该数组每一个子串的mex,记录一下,然后再求一次mex。因为数组规模是n,因此所有的mex必然是1~n+1的,所有可以枚举一下当前的i,判断当前的i是否可以作为mex,如果可以的话,记录一下i可以作为mex。那么如何判断i是否可以作为mex呢,对于一个已经a数组中存在的数字来说,只需要判断下1 ~ x - 1的所有数字的最近位置是否都在x上一次出现的位置的右边即可,如果都在说明x是一个mex值;对于一个不在a数组中存在的数字x来说,只需要判断下1 ~ x - 1的数字的最小下标是否为0即可。
代码:
#include <bits/stdc++.h>
using namespace std;
int const N = 1e5 + 10;
typedef long long LL;
typedef pair<int, int> PII;
int n, m, T;
int a[N], dat[N * 4], last[N];
set<int> s;
void pushup(int u) {
dat[u] = min(dat[u << 1], dat[u << 1 | 1]);
}
void modify(int u, int l, int r, int x, int y) {
if (l == x && r == x) {
dat[u] = y;
return;
}
int mid = (l + r) >> 1;
if (x <= mid ) modify(u << 1, l, mid, x, y);
else modify(u << 1 | 1, mid + 1, r, x, y);
pushup(u);
return;
}
int query(int u, int l, int r, int L, int R) {
if (L > R) return 0;
if (L <= l && r <= R) return dat[u];
int mid = (l + r) >> 1;
int res = 1e9 + 10;
if (L <= mid) res = min(res, query(u << 1, l, mid, L, R));
if (mid < R) res = min(res, query(u << 1 | 1, mid + 1, r, L, R));
return res;
}
int main() {
cin >> n;
for (int i = 1; i <= n; ++i) {
scanf("%d", &a[i]);
if (a[i] == 1) s.insert(2);
else s.insert(1);
}
for (int i = 1; i <= n; ++i) {
int x = a[i];
if (query(1, 1, n, 1, x - 1) > last[x]) {
s.insert(x);
}
modify(1, 1, n, x, i);
last[x] = i;
}
for (int i = 1; i <= n + 1; ++i)
if (last[i]) {
if (query(1, 1, n, 1, i - 1) > last[i]) s.insert(i);
}
else {
if (query(1, 1, n, 1, i - 1)) s.insert(i);
}
for (int i = 1; ; ++i)
if (!s.count(i)) {
cout << i << endl;
return 0;
}
return 0;
}