2021 MINIEYE杯 杭电多校1 1005 Minimum spanning tree 1006 Xor sum 01字典树

1005 Minimum spanning tree

题意：给定n-1个点，编号从2到n，两点a和b之间的边权为lcm(a, b)。找出它们形成的最小生成树。求该最小生成树的边权之和。

分析：对于任意两个不相等的点i，j，lcm(i, j) = i * j / gcd(i, j)。可以发现，对于质数i，其连接到2上时，边权最小，为i * 2；对于合数i，其连接到其任意因子上时边权最小，为i。所以最终答案为2~n内的每个质数乘2加上2~n内的每个合数。答案可以通过线性筛预处理求得。

Code：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10000005;
int t, n, v[maxn], pr[maxn], tot;
long long ans[maxn];

void init() {
    for (int i = 2; i < maxn; i++) {
        if (!v[i]) v[i] = i, pr[++tot] = i;
        for (int j = 1; j <= tot; j++) {
            if (v[i] < pr[j] || pr[j] > maxn / i) break;
            v[i * pr[j]] = pr[j];
        }
    }
    for (int i = 3; i < maxn; i++) 
        ans[i] = ans[i - 1] + i + i * (v[i] == i);
}

inline void solve() {
    scanf("%d", &n);
    printf("%lld\n", ans[n]);
}

int main() {
    init();
    scanf("%d", &t);
    while (t--) solve();
    return 0;
}

1006 Xor sum

题意：给定长度为n的序列，找出最短的异或和大于等于k的连续子序列。如果有多个相同长度的连续子序列，只输出左端点最小的连续子序列。如果不存在这样的子序列，则输出“-1”。

分析：首先获取序列的异或前缀和数组，则原问题转化为在异或前缀和数组中找到两个下标最近的数使得其异或和大于等于k。考虑维护一棵01字典树，在异或前缀和数组从头到尾的遍历过程中，字典树插入的值为异或和数组的前i-1个数，每个节点上再存扩展到当前节点的最大的下标（为了离i更近）。当遍历到第i个数m时，进行字典树搜索，如果有一个数与字典中的数异或结果的第k位大于m的第k位，那么该数与对应分支中所有的数异或结果都会大于m，更新答案，否则，就要在第k位异或相等的情况下，搜索更低位的异或结果。在搜索完字典树后，还需特殊考虑等于k的情况。最后将m插入字典树。

Code：

#include <bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 100005;
int t, n, k, ansl, ansr;
int a[N], tree[N * 30][2], cnt;
int maxl[N * 30]; // 扩展到当前结点的最大i

int main() {
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &k);
        for (int i = 1; i <= n; i++) {
            scanf("%d", a + i);
            a[i] ^= a[i - 1];
        }
        ansl = 0, ansr = n, cnt = 1;
        tree[1][0] = tree[1][1] = 0;
        maxl[1] = -1;
        for (int i = 0; i <= n; i++) {
            int p = 1, l = -1; // l满足a[i] ^ a[l] >= k，p为trie树索引
            for (int j = 29; j >= 0; j--) {
                int x = (a[i] >> j) & 1; // a[i]第j位为x
                if (!((k >> j) & 1)) { // k的第j位为0
                    if (tree[p][x ^ 1]) l = max(l, maxl[tree[p][x ^ 1]]); // 若存在l，使得(a[i] ^ a[l])第j位为1，此时a[i] ^ a[l] > k, 可以更新l
                    p = tree[p][x]; // 继续寻找可能满足条件的l，直到此时(a[i] ^ a[l])的第j~29位与k相同
                } else p = tree[p][x ^ 1]; // k的第j位为1，要满足a[i] ^ a[l] >= k，a[l]第j位只能为x ^ 1
                if (!p) break;
            }
            if (p) l = max(l, maxl[p]); // 特判，如果a[i] ^ a[l] == k
            if (l >= 0 && i - l < ansr - ansl + 1) ansl = l + 1, ansr = i;
            p = 1;
            for (int j = 29; j >= 0; j--) { // 向trie插入a[i]
                int x = (a[i] >> j) & 1;
                if (tree[p][x] == 0) {
                    tree[p][x] = ++cnt;
                    tree[cnt][0] = tree[cnt][1] = 0;
                    maxl[cnt] = -1;
                }
                p = tree[p][x];
                maxl[p] = max(maxl[p], i);
            }
        }
        if (ansl > 0) printf("%d %d\n", ansl, ansr);
        else puts("-1");
    }
    return 0;
}