求助!
#include<bits/stdc++.h>
using namespace std;
const int N=50005,LOG=20;
struct edge{
int to,last;
}e[N<<2];
int head[N<<2],cnt=0,depth[N],hop[N][LOG],sum[N],ans=0;
int n,k;
void add(int u,int v)
{
e[cnt].to=v;
e[cnt].last=head[u];
head[u]=cnt;
cnt++;
}
void dfs(int x,int f,int d)
{
depth[x]=d;
hop[x][0]=f;
int v;
for(int i=head[x];i;i=e[i].last)
{
v=e[i].to;
if(v==f)
{
continue;
}
dfs(v,x,d+1);
}
}
void init()
{
dfs(1,0,1);
for(int i=1;i<LOG;i++)
{
for(int j=1;j<=n;j++)
{
hop[j][i]=hop[hop[j][i-1]][i-1];
}
}
}
void du(int &u,int h)
{
for(int i=0;h;i++)
{
if(h&1)
{
u=hop[u][i];
}
h>>=1;
}
}
int lca(int u,int v)
{
if(depth[u]!=depth[v])
{
if(depth[u]<depth[v])
{
swap(u,v);
}
du(u,depth[u]-depth[v]);
}
if(u==v)
{
return u;
}
for(int i=LOG-1;i>=0;i++)
{
if(hop[u][LOG]==hop[v][LOG])
{
continue;
}
u=hop[u][i];
v=hop[v][i];
}
return hop[u][0];
}
void query(int x,int f)
{
int v;
for(int i=head[x];i;i=e[i].last)
{
v=e[i].to;
if(v==f)
{
continue;
}
query(v,x);
sum[x]+=sum[v];
}
ans=max(ans,sum[x]);
}
int main()
{
scanf("%d%d",&n,&k);
int u,v;
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
init();
int l;
for(int i=1;i<=k;i++)
{
scanf("%d%d",&u,&v);
l=lca(u,v);
sum[u]++;
sum[v]++;
sum[l]--;
if(l>1)
{
sum[hop[l][0]]--;
}
}
query(1,0);
printf("%d",ans);
return 0;
}
| 测试点编号 | 评测结果 |
|---|---|
| #1 | AC |
| #2 | TLE |
| #3 | TLE |
| #4 | TLE |
| #5 | TLE |
| #6 | TLE |
| #7 | TLE |
| #8 | TLE |
| #9 | TLE |
| #10 | TLE |
| #11 | TLE |
| #12 | TLE |
| #13 | TLE |
| #14 | TLE |
| #15 | TLE |