Codeforces Round #778 D

Codeforces Round #778 (Div. 1 + Div. 2, based on Technocup 2022 Final Round)

Codeforces Round #778 (Div. 1 + Div. 2, based on Technocup 2022 Final Round)

阳间场好像就没上过分，但掉这么多是我没想到的

D. Potion Brewing Class

prob. ：给一棵树，每个点有个权值，但不知道具体是多少，信息是对于每一条树边给出两个端点权值的比值，要求所有点权都为正整数，求所有点权总和的最小值。

idea：

考虑取一个点为根，将这个点的权值设为"单位1"， 然后dfs对于每个点用分数表示它的点权，再通分。
分数和通分部分考虑对于每个质因子存一个次数同时维护每个质因子的最高次数，然后更改单位1"的值，重新dfs求每个点权值。

---

有一段时间在想这样一个问题，可以说是自己把自己绕死了：

对于每条边两端的点，比如A和B，假设$\frac{A}{B}=\frac{x}{y}$ , 我认为A是x的倍数会导致整个以A为根B为父节点的子树全是x的倍数，即AB这条边的比值会影响到C

然后一直对为什么能直接dfs想不明白 （怎么有人

但这里第一遍的时候其实全是分数形式，在后面对根的值变化的时候会把这个x的部分包括进去

code ：

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll mod = 998244353;
const ll N = 2e5 + 10;
ll primes[N], cnt = 0;
bool st[N];
vector<ll> fac[N];

void sieve() {
    cnt = 0;
    for (ll i = 2; i < N; ++i) {
        if (!st[i]) primes[++cnt] = i;
        for (ll j = 1; primes[j] * i < N; ++j) {
            st[primes[j] * i] = 1;
            if (i % primes[j] == 0) break;
        }
    }
}

void getFactor() {
    for (ll i = 2; i < N; ++i) {
        ll x = i;
        for (ll j = 1; primes[j] <= x / primes[j]; ++j) {
            while (x % primes[j] == 0) {
                fac[i].push_back(primes[j]);
                x /= primes[j];
            }
        }
        if (x > 1) fac[i].push_back(x);
    }
}

struct node {
    ll to, x, y;
};

vector<node> g[N];
ll f[N], mxf[N];

void dfsGetFrac(ll from, ll p) {
    for (auto tmpNode : g[from]) {
        ll to = tmpNode.to, x = tmpNode.x, y = tmpNode.y;
        if (to == p) continue;
        for (ll num :fac[x]) f[num]++, mxf[num] = max(mxf[num], f[num]);
        for (ll num : fac[y]) f[num]--;
        dfsGetFrac(to, from);
        for (ll num :fac[x]) f[num]--;
        for (ll num : fac[y]) f[num]++;
    }
}

ll qpow(ll x, ll n) {
    ll res = 1;
    while (n) {
        if (n & 1) res = res * x % mod;
        x = x * x % mod;
        n >>= 1;
    }
    return res;
}

ll ans = 0;

void dfs(ll from, ll p, ll res) {
    ans = (ans + res) % mod;
    for (auto tmpNode: g[from]) {
        ll to = tmpNode.to, x = tmpNode.x, y = tmpNode.y;
        if (to == p) continue;
        ll tmp = res;
        for (ll num : fac[x]) tmp = tmp * qpow(num, mod - 2) % mod;
        for (ll num : fac[y]) tmp = tmp * num % mod;
        dfs(to, from, tmp);
    }
}

void init(ll n) {
    for (ll i = 0; i <= n; ++i) {
        f[i] = mxf[i] = 0;
        g[i].clear();
    }
    ans = 0;
}


signed main() {
    ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);

    sieve();
    getFactor();

    ll T;
    cin >> T;
    while (T--) {
        ll n;
        cin >> n;
        for (ll i = 0; i < n - 1; ++i) {
            ll u, v, x, y;
            cin >> u >> v >> x >> y;
            ll d = __gcd(x, y);
            x /= d;
            y /= d;
            g[u].push_back({v, x, y});
            g[v].push_back({u, y, x});
        }

        dfsGetFrac(1, -1);
        ll p = 1;
        for (ll i = 1; i <= n; ++i) {
            for (ll j = 0; j < mxf[i]; ++j) {
                p = p * i % mod;
            }
        }
        cerr << p << endl;
        dfs(1, -1, p);
        cout << ans << endl;
        init(n);
    }
    return 0;
}

E. Arithmetic Operations

prob. ：给一个数列，一次改变可以将任意位置的一个数变成任意想要的数，要求最少改变次数使得数列变成等差

idea ：

要求改变最少就是留下最多

以数列下标为x轴，以数值为y轴转化为二维平面上n个点，找出覆盖点最多的直线

直线由斜率和与y轴的交点唯一确定

考虑从斜率（公差d）入手

至少两个点在我们要找的这条线上，考虑到$a_i$的范围，斜率（公差d）也有个范围

分类

1. 当$d<\sqrt{N}$ ， 直接对于每个点枚举d，把和y轴的交点存到对应的hashmap （每个点是$\sqrt{N}$，总体$N\sqrt{N}$
2. 当$d\ge \sqrt{N}$， 枚举上一个点出现的位置，求出斜率和与y轴的坐标，计数 从这个区间内找值转移，（因为$a_i$ 的大小限制，当d大于$\sqrt{N}$时 ，在同一条线上的下标差值不会超过$\sqrt{N}$ ）设$d{p}_{i,d}$ 表示到i为止以d为斜率同时i在线上时线上的点最多是多少（每个点是$\sqrt{N}$，总体$N\sqrt{N}$

这两部分都可以用一个类似hashmap去计数， 因为d为整数，与y轴的交点一定是整数

斜率为负数的时候，将整个数组reverse再做一遍

---

upd：unordered_map 会T

upupd：我真的会谢，感觉手写hash容易撞

对不起我刚开始还觉得官方题解建什么图啊d什么p啊，现在觉得官方题解写的很好（

dp部分可以直接map，因为log大概是threshold的大小，所以常数不大

---

hash那个想法好像能用vector过