noi.ac #240 tree

说实话只要明白了题意这个题还是比较水的。。。（样例我差点蒙了。。

具体就是先把每个点的点权都平方计入答案（因为每个点也算一条路径。。）

再按题意模拟就行了。

我的 \(dfs\) 是对每个点进行的，因此一条两点之间的路径被计算了两次。所以要除以 \(2\)。

nmd这取模是真的烦，不过还是找到了。对路径上的答案的取模要单独分开，不能和单独的点的答案一起计算，即不能写成

printf("%lld\n", ((ans / 2) + sum + mod) % mod);

而应该写成

printf("%lld\n", (ans * (mod / 2 + 1) + sum + mod) % mod);

或者乘2的逆元也行，要写快速幂。。

printf("%lld\n", (ans * ksm(2,mod-2,mod) + sum + mod) % mod);

我一开始没想起来我透。。

在此orz女开神

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll N = 1e5 + 10;
const ll mod = 998244353;
ll head[N], Next[N<<1], ver[N<<1], tot = 1;
ll val[N];
template<class I>
inline void rd(I &x) {
    ll f = 1;char c = getchar();
    for(;c < '0' || c > '9';c = getchar()) if(c == '-') f = -1;
    for(x = 0;c >= '0' && c <= '9';x = (x << 3) + (x << 1) + (c & 15), c = getchar());
    x *= f;
}
void add(ll x, ll y) {
    ver[++tot] = y;
    Next[tot] = head[x];
    head[x] = tot;
}
ll fa[N][12];
queue<ll> q;
ll d[N];
ll ans;
void dfs(ll x, ll fa, ll maxx, ll minn) {
    maxx = max(maxx, val[x]);
    minn = min(minn, val[x]);
    ans = (ans % mod + maxx * minn % mod + mod) % mod;
    for (ll i = head[x];i;i = Next[i]) {
        ll y = ver[i];
        if(y == fa) continue;
        dfs(y,x,maxx,minn);
    }
}
ll n;
ll sum;
int main() {
    rd(n);
    for (ll i = 1;i <= n;i++) {
        rd(val[i]);
        sum = (sum % mod + val[i] * val[i] % mod + mod) % mod;
    }
    for (ll i = 1;i < n;i++) {
        ll x, y;
        rd(x), rd(y);
        add(x,y);
        add(y,x);
    }
    for (ll x = 1;x <= n;x++)
        for (ll i = head[x];i;i = Next[i]) {
            ll y = ver[i];
            dfs(y,x,val[x],val[x]);
        }
    printf("%lld\n", (ans * (mod / 2 + 1) + sum + mod) % mod);
    return 0;
} 

树上的父亲真好用