【Hihocoder 1167】 高等理论计算机科学 （树链的交，线段树或树状数组维护区间和）

6 3
1 2
2 3
2 4
4 5
4 6
1 3
1 5
5 6

2

1 #include<cstdio>
2 #include<cstdlib>
3 #include<cstring>
4 #include<iostream>
5 #include<algorithm>
6 #include<cmath>
7 using namespace std;
8 #define Maxn 100010
9 #define LL long long
10
11 struct node
12 {
13     int x,y,next;
14 }t[Maxn*2];int len;
15
16 int first[Maxn],px[Maxn],py[Maxn];
17
18 void ins(int x,int y)
19 {
20     t[++len].x=x;t[len].y=y;
21     t[len].next=first[x];first[x]=len;
22 }
23
24 int son[Maxn],dfn[Maxn],sm[Maxn],dep[Maxn],fa[Maxn];
25 void dfs1(int x,int f)
26 {
27     sm[x]=1;son[x]=0;dep[x]=dep[f]+1;fa[x]=f;
28     for(int i=first[x];i;i=t[i].next) if(t[i].y!=f)
29     {
30         int y=t[i].y;
31         dfs1(y,x);
32         sm[x]+=sm[y];
33         if(sm[y]>sm[son[x]]) son[x]=y;
34     }
35 }
36
37 int tp[Maxn],cnt;
38 void dfs2(int x,int f,int tpp)
39 {
40     dfn[x]=++cnt;tp[x]=tpp;
41     if(son[x]) dfs2(son[x],x,tpp);
42     for(int i=first[x];i;i=t[i].next) if(t[i].y!=f&&t[i].y!=son[x])
43      dfs2(t[i].y,x,t[i].y);
44 }
45
46 int c[Maxn],n;
47 bool lca[Maxn];
48
49 void add(int x,int y)
50 {
51     for(int i=x;i<=n;i+=i&(-i))
52         c[i]+=y;
53 }
54
55 int query(int l,int r)
56 {
57     int ans=0;
58     for(int i=r;i>=1;i-=i&(-i))
59         ans+=c[i];
60     l--;
61     for(int i=l;i>=1;i-=i&(-i))
62         ans-=c[i];
63     return ans;
64 }
65
66 int gans(int x,int y,int p)
67 {
68     int ans=0,tt;
69     while(tp[x]!=tp[y])
70     {
71         if(dep[tp[x]]<dep[tp[y]]) tt=x,x=y,y=tt;
72         if(p==1) ans+=query(dfn[tp[x]],dfn[x]);
73         x=fa[tp[x]];
74     }
75     if(dep[x]<dep[y]) tt=x,x=y,y=tt;
76     if(p==1)
77     {
78         ans+=query(dfn[y],dfn[x]);
79         return ans;
80     }
81     else return y;
82 }
83
84 int main()
85 {
86     int p;
87     LL ans=0;
88     scanf("%d%d",&n,&p);
89     len=0;
90     memset(first,0,sizeof(first));
91     for(int i=1;i<n;i++)
92     {
93         int x,y;
94         scanf("%d%d",&x,&y);
95         ins(x,y);ins(y,x);
96     }
97     for(int i=1;i<=p;i++) scanf("%d%d",&px[i],&py[i]);
98     sm[0]=0;dep[0]=0;
99     dfs1(1,0);cnt=0;
100     dfs2(1,0,1);
101     memset(c,0,sizeof(c));
102     memset(lca,0,sizeof(lca));
103     for(int i=1;i<=p;i++)
104     {
105         int x=gans(px[i],py[i],0);
106         lca[x]=1;
107         add(dfn[x],1);
108     }
109     for(int i=1;i<=p;i++)
110     {
111         int x=gans(px[i],py[i],1);
112         ans+=x;
113     }
114     for(int i=1;i<=n;i++) if(lca[i])
115     {
116         int x=query(dfn[i],dfn[i]);
117         ans-=x*(x-1)/2+x;
118     }
119     printf("%lld\n",ans);
120     return 0;
121 }