树的储存与遍历 -- 2019暑期集训第2次校赛-E. 树和森林

2019暑期集训第2次校赛-E. 树和森林
链接：https://ac.nowcoder.com/acm/contest/1076/E
来源：牛客网

题目描述
有一片森林，森林中有m颗树，每棵树大小都是n且结构都一模一样，由题目给出，现在你有m - 1次操作，每次操作将第a颗树的u号点和第b颗树的v号点连接（保证他们之前是不连通的），求m次操作后整颗树所有点对距离和，答案对998244353取模
输入描述:
第一行两个正整数n,m(1≤n,m≤1000)

接下来n−1行，每行两个正整数u,v(1≤u,v≤n)，表示初始时每棵树的边

接下来m−1行，每行四个正整数a,b,u,v(1 <= a, b <= m, 1 <= u, v <= n)
输出描述:
输出一个数表示答案
示例1
输入
复制
3 2
1 2
2 3
1 2 1 1
输出
复制
35
思路
算任意两点间距离之和，可以转换为每条边被多少区间包含，则这条边对就答案贡献了多少次。具体实现为把树用vector存起来。再dfs遍历树求解。
题解

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1005,mod=998244353;
vector<int>G[1000001];
int d[maxn*maxn],n,m;
ll ans=0;
void dfs(int u,int fa)
{
    d[u]=1;
    for(int i=0;i<G[u].size();i++)                     //dfs遍历树
    {
        int v=G[u][i];
        if(v==fa)continue;
        dfs(v,u);                       
        d[u]+=d[v];                                                              
    }
    ans=(ans+1ll*d[u]*(n*m-d[u])%mod)%mod;    
}
int main()
{
    int a,b,u,v;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        for(int j=1;j<=m;j++)
        {
            int uu=u+(j-1)*n;  //存每颗小树
            int vv=v+(j-1)*n;
            G[uu].push_back(vv);
            G[vv].push_back(uu);
        }
    }
    for(int i=1;i<m;i++)
    {
        scanf("%d%d%d%d",&a,&b,&u,&v);
        int uu=u+(a-1)*n;   //存连接后的大树
        int vv=v+(b-1)*n;
        G[uu].push_back(vv);
        G[vv].push_back(uu);
    }
    dfs(1,0);
    printf("%lld\n",ans);
}