D. Tree Problem（dfs统计子树节点数）

优化一下公式后发现只需求每棵子树中点的数量即可，记录一个重要模板，dfs统计子树的节点数

记录一个重要优化公式的方式对以下求和

int sum=0;
for(int i=1;i<=n;i++)
{
    for(int j=i+1;j<=n;j++)
    {
        sum+=a[i][j];
    }
}

我们可以将上三角形的和转化为整个矩阵的和，减去对角线的平方的和再除以2

对于本题，我们优化后就不需要枚举每两棵子树的大小的乘积，达到优化时间复杂度的目的

#include <bits/stdc++.h>
using namespace std;
void TLE() { ios::sync_with_stdio(false), cin.tie(0), cout.tie(0); }
#define int long long
typedef long long ll;
typedef unsigned long long ULL;
const int N = 2e5 + 10;
const ll mod = 1000000007;
int sz[N];
signed main()
{
    int n;
    cin >> n;
    vector<vector<int>> edge(n + 1);
    vector<int> dp(n + 1);
    for (int i = 1; i <= n - 1; i++)
    {
        int u, v;
        cin >> u >> v;
        edge[u].push_back(v);//本题为无向边
        edge[v].push_back(u);
    }
    function<void(int,int)> dfs = [&](int u, int fa)
    {
        sz[u] = 1;
        int sum = 0;
        for (auto v : edge[u])
        {
            if (v == fa)
            {
                continue;
            }
            dfs(v, u);
            sz[u] += sz[v];
            sum += sz[v] * sz[v];
        }
        sum += (n - sz[u]) * (n - sz[u]);
        dp[u] = ((n - 1) * (n - 1) - sum) / 2;
    };
    dfs(1, -1);
    int q;
    cin >> q;
    while (q--)
    {
        int u;
        cin >> u;
        cout << dp[u] + (n - 1) << endl;
    }

    return 0;
}