Kuro is living in a country called Uberland, consisting of n
).
Oddly, there are 2 special towns in Uberland named Flowrisa (denoted with the index x
) and Beetopia (denoted with the index y). Flowrisa is a town where there are many strong-scent flowers, and Beetopia is another town where many bees live. In particular, Kuro will avoid any pair of towns (u,v) if on the path from u to v, he reaches Beetopia after he reached Flowrisa, since the bees will be attracted with the flower smell on Kuro’s body and sting him.
Kuro wants to know how many pair of city (u,v)
he can take as his route. Since he’s not really bright, he asked you to help him with this problem.
The first line contains three integers n
) - the number of towns, index of the town Flowrisa and index of the town Beetopia, respectively.
n−1
lines follow, each line contains two integers a and b ( 1≤a,b≤n, a≠b), describes a road connecting two towns a and b.
It is guaranteed that from each town, we can reach every other town in the city using the given roads. That is, the given map of towns and roads is a tree.
A single integer resembles the number of pair of towns (u,v)
that Kuro can use as his walking route.
3 1 3 1 2 2 3
5
3 1 3 1 2 1 3
4
On the first example, Kuro can choose these pairs:
Kuro can't choose pair (1,3)
since his walking route would be 1→2→3, in which Kuro visits town 1 (Flowrisa) and then visits town 3 (Beetopia), which is not allowed (note that pair (3,1)is still allowed because although Kuro visited Flowrisa and Beetopia, he did not visit them in that order).
On the second example, Kuro can choose the following pairs:
#include
#include
#include
#include
#include
#include
#include
using namespace std;
long long n,x,y;
vector > mp(300005, vector(0,0));
int ans=0;
int vis[300005]={};
long long cntx=0,cnty=0;
int findy(int cur){///找到x到y的路径,并标记路径上的点(包括x,y)vis=1
if(cur==y){
vis[cur]=1;
return 1;
}
int next;
vis[cur]=-1;
for(int i=0; i>n>>x>>y;
int a,b;
for(int i=0;i>a>>b;
mp[a].push_back(b);
mp[b].push_back(a);
}
findy(x);
calx(x);
caly(y);
long long ans=n*(n-1)-cntx*cnty;
cout<
要注意数据范围,用int溢出。使用longlong;
使用排除法,n个顶点任取两个做顶点对,有n*n-1个,减去不符合条件的即为答案。
把点分为三类:在x到y路径上的,挂在x上不与y通的,计数cntx个,挂在y上不与x通的,计数cnty个。
所有(第二类,第三类)的点对都是不符合条件的,所以ans=n*(n-1) - cntx*cnty;
*****学会了用vector实现动态二维数组:
初始化方式如下
vector > mp(300005, vector(0,0));
--注意后面两个>>中间要加空格
向数组中加入元素方式如下
mp[a].push_back(b);
mp[b].push_back(a);