352. 闇の連鎖

传说中的暗之连锁被人们称为 Dark。

Dark 是人类内心的黑暗的产物，古今中外的勇者们都试图打倒它。

经过研究，你发现 Dark 呈现无向图的结构，图中有 N 个节点和两类边，一类边被称为主要边，而另一类被称为附加边。

Dark 有 N–1 条主要边，并且 Dark 的任意两个节点之间都存在一条只由主要边构成的路径。

另外，Dark 还有 M 条附加边。

你的任务是把 Dark 斩为不连通的两部分。

一开始 Dark 的附加边都处于无敌状态，你只能选择一条主要边切断。

一旦你切断了一条主要边，Dark 就会进入防御模式，主要边会变为无敌的而附加边可以被切断。

但是你的能力只能再切断 Dark 的一条附加边。

现在你想要知道，一共有多少种方案可以击败 Dark。

注意，就算你第一步切断主要边之后就已经把 Dark 斩为两截，你也需要切断一条附加边才算击败了 Dark。

输入格式

第一行包含两个整数 N 和 M。

之后 N–1 行，每行包括两个整数 A 和 B，表示 A 和 B 之间有一条主要边。

之后 M 行以同样的格式给出附加边。

输出格式

输出一个整数表示答案。

数据范围

N≤100000,M≤200000，数据保证答案不超过231−1

输入样例：

4 1
1 2
2 3
1 4
3 4

输出样例：

3

 

分析：先考虑主要边，删除一条主要边后，不考虑附加边的话，整棵树一定会被分成两块。

再考虑附加边。如果没有附加边连在这两块之间的话，那删除任意一条附加边都可以。总贡献为m

如果恰好有1条附加边连在这两块之间的话，那我们一定要删除这条附加边,总贡献为1。如果有多于1条附加边连在这两块之间的话，那因为只能再删一条附加边，我们没办法切成这两块了。总贡献为0所以问题就变成了，如何快速查询两块之间有多少条附加边。可以在每一个附加边了的两个端点a,b,分别d[a]++,d[b]++，两个点的最近公共祖先u就d[u]-=2，这样就可以根据该子树端点d的和来判断子树中有几个连向外面的附加边。

代码如下：

#include <iostream>
#include <vector>
using namespace std;

const int N=1e5+10;
int depth[N],lg[N],fa[N][20],d[N];
vector<int>adj[N];
int n,m,ans;

void dfs(int f,int u)
{
    depth[u]=depth[f]+1;
    //cout<<u<<" u "<<depth[u]<<endl;
    int d=depth[u];
    fa[u][0]=f;
    for(int i=1;i<=lg[d];i++)
        fa[u][i]=fa[fa[u][i-1]][i-1];
    for(auto v:adj[u])
    {
        if(v==f)continue;
        dfs(u,v);
    }
}
void init()
{
    for(int i=1;i<=n;i++)
        lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    for(int i=1;i<=n;i++) lg[i]--;
    dfs(0,1);
}
int lca(int x,int y)
{
    if(depth[x]<depth[y])swap(x,y);
    while(depth[x]>depth[y])
        x=fa[x][lg[depth[x]-depth[y]]];
    if(x==y) return x;
    for(int i=lg[depth[x]];i>=0;i--)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int dfs1(int f,int u)
{
    int res=d[u];
    for(auto v:adj[u])
    {
        if(v==f)continue;
        int tem=dfs1(u,v);
        if(tem==0)ans+=m;
        else if(tem==1)ans+=1;
        res+=tem;
    }
    return res;
}

int main()
{
    cin>>n>>m;
    
    for(int i=1;i<n;i++)
    {
        int x,y;
        cin>>x>>y;
        adj[x].push_back(y);
        adj[y].push_back(x);
    }
    init();
    for(int i=1;i<=m;i++)
    {
        int x,y;
        cin>>x>>y;
        int f=lca(x,y);
        d[x]++,d[y]++,d[f]-=2;
    }
    dfs1(-1,1);
    cout<<ans;
    return 0;
}