3910: 火车
Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 262 Solved: 90
[Submit][Status][Discuss]
Description
A 国有n 个城市,城市之间有一些双向道路相连,并且城市两两之间有唯一路径。现在有火车在城市 a,需要经过m 个城市。火车按照以下规则行驶:每次行驶到还没有经过的城市中在 m 个城市中最靠前的。现在小 A 想知道火车经过这m 个城市后所经过的道路数量。
Input
第一行三个整数 n、m、a,表示城市数量、需要经过的城市数量,火车开始时所在位置。
接下来 n-1 行,每行两个整数 x和y,表示 x 和y之间有一条双向道路。
接下来一行 m 个整数,表示需要经过的城市。
Output
一行一个整数,表示火车经过的道路数量。
Sample Input
5 4 2
1 2
2 3
3 4
4 5
4 3 1 5
Sample Output
9
HINT
N<=500000 ,M<=400000
Source
Solution
水题- -最多算个并查集的有趣应用
很显然直接询问用LCA统计答案即可
至于处理走过的路径,拿并查集维护一下,很简单的把起止点到LCA的点合并一下,询问的两个点如果属于一个集合显然走过
Code
#include<iostream> #include<cmath> #include<cstdio> #include<algorithm> #include<cstring> using namespace std; int read() { int x=0,f=1; char ch=getchar(); while (ch<‘0‘ || ch>‘9‘) {if (ch==‘-‘) f=-1; ch=getchar();} while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();} return x*f; } #define maxn 501000 int n,m,a; struct EdgeNode{int to,next;}edge[maxn<<1]; int head[maxn],cnt; void add(int u,int v) {cnt++; edge[cnt].next=head[u]; head[u]=cnt; edge[cnt].to=v;} void insert(int u,int v) {add(u,v); add(v,u);} int deep[maxn],father[maxn][20],ffff[maxn]; long long ans; void dfs(int now) { for (int i=1; i<=19; i++) if (deep[now]>=(1<<i)) father[now][i]=father[father[now][i-1]][i-1]; else break; for (int i=head[now]; i; i=edge[i].next) if (edge[i].to!=father[now][0]) { deep[edge[i].to]=deep[now]+1; father[edge[i].to][0]=now; dfs(edge[i].to); } } int LCA(int x,int y) { if (deep[x]<deep[y]) swap(x,y); int dd=deep[x]-deep[y]; for (int i=0; (1<<i)<=dd; i++) if (dd&(1<<i)) x=father[x][i]; for (int i=19; i>=0; i--) if (father[x][i]!=father[y][i]) x=father[x][i],y=father[y][i]; if (x==y) return x; return father[x][0]; } int find(int x) {if (ffff[x]==x) return x; ffff[x]=find(ffff[x]); return ffff[x];} int ff1,ff2; int main() { n=read(),m=read(),a=read(); for (int u,v,i=1; i<=n-1; i++) u=read(),v=read(),insert(u,v); dfs(1); for (int i=1; i<=n; i++) ffff[i]=i; for (int i=1; i<=m; i++) { int x=read(); int fa=find(a),fx=find(x); if (fa==fx) continue; int lca=LCA(a,x); ans+=deep[a]-deep[lca]+deep[x]-deep[lca]; int ta=a,tx=x,flca; flca=find(lca); while (find(ta)!=flca) {ff1=find(ta); ffff[ff1]=flca; ta=father[ff1][0];} while (find(tx)!=flca) {ff2=find(tx); ffff[ff2]=flca; tx=father[ff2][0];} a=x; } cout<<ans; return 0; }
电脑炸出奇怪的错误,所以写的比较鬼畜- -
时间: 2024-10-19 00:04:13