比赛 新年快乐 评测结果 AAATTTTTTTTTTTTTTTTT
题目名称 收益 最终得分 15
用户昵称 梦那边的美好BP 运行时间 53.689 s
代码语言 C++ 内存使用 270.66 MiB
提交时间 2026-02-13 11:26:35
显示代码纯文本
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int N = 4e6 + 6;
typedef long long ll;
ll a[N];
int head[N], nxt[N], to[N], num = 1;
void add(int u, int v) {
    num++;
    to[num] = v;
    nxt[num] = head[u];
    head[u] = num;
}

int sz[N], dep[N], fa[N], son[N], top[N], id[N], rnk[N], tot = 0;
void dfs1(int u) {
    sz[u] = 1;
    for (int i = head[u]; i; i = nxt[i]) {
        int v = to[i];
        dep[v] = dep[u] + 1;
        fa[v] = u;
        dfs1(v);
        sz[u] += sz[v];
        if (sz[v] > sz[son[u]]) {
            son[u] = v;
        }
    }
}
void dfs2(int u, int ance) {
    top[u] = ance;
    id[u] = ++tot;
    rnk[tot] = u;
    if (!son[u]) return;
    dfs2(son[u], ance);
    for (int i = head[u]; i; i = nxt[i]) {
        int v = to[i];
        if (v != son[u]) {
            dfs2(v, v);
        }
    }
}

#define lc (u * 2)
#define rc (u * 2 + 1)
#define mid ((l + r) / 2)
ll tr[N * 4], lazy[N * 4];
void build(int u, int l, int r) {
    if (l == r) {
        tr[u] = a[rnk[l]];
        return;
    }
    build(lc, l, mid);
    build(rc, mid + 1, r);
    tr[u] = tr[lc] + tr[rc];
}
void lz(int u, int l, int r, ll k) {
    lazy[u] = k;
    tr[u] = k * (r - l + 1);
}
void pd(int u, int l, int r) {
    if (lazy[u] == -1) return;
    lz(lc, l, mid, lazy[u]);
    lz(rc, mid + 1, r, lazy[u]);
    lazy[u] = -1;
}
ll query(int u, int l, int r, int x, int y) {
    if (y < l || x > r) return 0;
    if (x <= l && r <= y) {
        return tr[u];
    }
    pd(u, l, r);
    return query(lc, l, mid, x, y) + query(rc, mid + 1, r, x, y);
}
void change(int u, int l, int r, int x, int y, ll k) {
    if (x <= l && r <= y) {
        lz(u, l, r, k);
        return;
    }
    pd(u, l, r);
    if (x <= mid) change(lc, l, mid, x, y, k);
    if (mid < y) change(rc, mid + 1, r, x, y, k);
    tr[u] = tr[lc] + tr[rc];
}
int n, k, sed;
void cg(int x, int y, int z) {
    while (top[x] != top[y]) {
        if (dep[top[x]] < dep[top[y]]) {
            swap(x, y);
        }
        change(1, 1, n, id[top[x]], id[x], z);
        x = fa[top[x]];
    }
    if (dep[x] > dep[y]) {
        swap(x, y);
    }
    change(1, 1, n, id[x], id[y], z);
}
ll ask(int x, int y) {
    ll ans = 0;
    while (top[x] != top[y]) {
        if (dep[top[x]] < dep[top[y]]) {
            swap(x, y);
        }
        ans = ans + query(1, 1, n, id[top[x]], id[x]);
        x = fa[top[x]];
    }
    if (dep[x] > dep[y]) {
        swap(x, y);
    }
    ans = ans + query(1, 1, n, id[x], id[y]);
    return ans;
}
bool b[N];
vector<int> v;

// ll sum[N];
// void dfs(int x, ll y) {
//     y += a[x];
//     sum[x] = y;
//     for (int i = head[x]; i; i = nxt[i]) {
//         int v = to[i];
//         dfs(v, y);
//     }
// }
int main() {
    freopen("x.in", "r", stdin);
    freopen("x.out", "w", stdout);
    cin >> n >> k >> sed;
    a[1] = sed;
    for (int i = 2; i <= n; i++) {
        a[i] = (a[i - 1] * 23333333 + 6666666) % 1000000007;
        int f = (a[i] ^ 23333333) % (i - 1) + 1;
        add(f, i);
        b[f] = 1;
    }
    for (int i = 1; i <= n; i++) {
        if (!b[i]) v.push_back(i);
    }
    dfs1(1);
    dfs2(1, 1);
    ll ans = 0;
    // while (k--) {
    //     ll len = v.size(), id = 0, mx = -1;
    //     dfs(1, 0);
    //     for (int i = 0; i <= len - 1; i++) {
    //         if (sum[v[i]] > mx) {
    //             id = v[i];
    //             mx = sum[v[i]];
    //         }
    //     }
    //     ans += mx;
    //     while (id) {
    //         a[id] = 0;
    //         id = fa[id];
    //     }
    // }
    // cout << ans;

    // return 0;
    memset(lazy, -1, sizeof(lazy));
    build(1, 1, n);
    while (k--) {
        ll len = v.size(), id = 0, mx = -1;
        for (int i = 0; i <= len - 1; i++) {
            ll cnt = ask(1, v[i]);
            if (cnt > mx) {
                id = v[i];
                mx = cnt;
            }
        }
        ans += mx;
        cg(1, id, 0);
    }
    cout << ans;
    return 0;
}