树上任意两点（距离，权值等）之和相关问题

吉首大学2019年程序设计竞赛- 滑稽树下你和我
 

题意：给出一棵树，树上每条边给出一个权值，现在要求（distance(i,j)表示i到j节点的距离）。

题解：这种题肯定不能是暴力，现在我们就可以看每条边最终要被计算多少次，这样我们只需要我们只需要更新边的贡献值就行了，看下图。

 现在我们来计算(u,v)之间的边的贡献值，现加入v为u的子节点（建双向边，任一点都可以为根节点，现在就假如我们这样简图），我们首先跑一遍DFS，将每个节点下面有多少个子节点计算出来，用sum数组保存，现在sum[u]=5,sum[v]=4,这个边的贡献值就是有多少条路径要经过他，很显然是sum[v]乘以(n-sum[v])，（n表示有n个节点）。

#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<cmath>
#include<stack>
#include<string>

const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
const int inf = 1e9;
const long long onf = 1e18;
#define me(a, b) memset(a,b,sizeof(a))
#define lowbit(x) x&(-x)
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI 3.14159265358979323846
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int head[maxn << 1], tot = 0;
ll sum[maxn];
ll ans = 0;
int n;
struct node {
    int v, next;
    ll w;
} maps[maxn << 1];

void add_edge(int u, int v, ll w) {
    maps[tot].v = v;
    maps[tot].w = w;
    maps[tot].next = head[u];
    head[u] = tot++;
}

void dfs(int u, int fa) {
    sum[u] = 1;
    for (int i = head[u]; ~i; i = maps[i].next) {
        int v = maps[i].v;
        if (v == fa)
            continue;
        dfs(v, u);
        sum[u] += sum[v];
    }
}

void solve(int u, int fa) {
    for (int i = head[u]; ~i; i = maps[i].next) {
        int v = maps[i].v;
        ll w = maps[i].w;
        if (v == fa)
            continue;
        ans = (ans + (ll) sum[v] * (n - sum[v]) * w) % mod;
        solve(v, u);
    }
}

void init() {
    me(head, -1);
    tot = 0;
}

int main() {
    init();
    scanf("%d", &n);
    for (int i = 1; i < n; i++) {
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        add_edge(u, v, w);
        add_edge(v, u, w);
    }
    dfs(1, -1);
    solve(1, -1);
    printf("%lld\n", ans);
    return 0;
}

2019长安大学ACM校赛-LaTale

题意：给出一棵树，给出树上每条边的权值，现在让你找出一对节点要求两节点的距离能够被三整除，问这棵树上能找出多少个这样的节点。

题解：枚举每个点，计算出以这个点为要找的两个点左边那个点时（左边为相对），满足条件的另一个点有多少个。

#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<cmath>
#include<stack>
#include<string>

const int mod = 998244353;
const int maxn = 1e5 + 5;
const int inf = 1e9;
const long long onf = 1e18;
#define me(a, b) memset(a,b,sizeof(a))
#define lowbit(x) x&(-x)
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI 3.14159265358979323846
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int head[maxn],tot=0;
int Next[maxn<<1],mapsw[maxn<<1],mapsv[maxn<<1];
int n;
ll ans=0;
ll dp[maxn][3];
void add_edge(int u,int v,int w){
    mapsw[tot]=w;
    mapsv[tot]=v;
    Next[tot]=head[u];
    head[u]=tot++;
}
void init(){
    me(head,-1);
    tot=0;
}
void dfs(int u,int fa){
    dp[u][0]=1;
    for(int i=head[u];~i;i=Next[i]){
        int v=mapsv[i];
        int w=mapsw[i];
        if(v==fa)
            continue ;
        dfs(v,u);
        ans+=dp[u][0]*dp[v][(0-w+3)%3];///因为u,v之间的长度为w，所以想要与v距离为(0-w+3)%3的，u到该点的距离才能被三整除。
        ans+=dp[u][1]*dp[v][(2-w+3)%3];
        ans+=dp[u][2]*dp[v][(1-w+3)%3];
        dp[u][(0+w)%3]+=dp[v][0];///要是v与改些点的距离为0，加上u到v的距离w，u到该点的距离就为（0+w）%3
        dp[u][(1+w)%3]+=dp[v][1];
        dp[u][(2+w)%3]+=dp[v][2];
    }
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) {
        init();
        scanf("%d",&n);
        for(int i=1;i<n;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w%3);
            add_edge(v,u,w%3);
        }
        ans=0;
        me(dp,0);
        dfs(1,-1);
        printf("%lld\n",ans);
    }
    return 0;
}