又来重新写了一遍运输计划
不得不说noip2015的分量还是很重的orz
比上次写多了5分,可是最后一个测试点怎么也得要2.4s不开心(打滚qwq)
贴代码吧~
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=3e5+5;
int n,m,plan[N][4],a,b,c,fa[N][20],wor[N][20],dp[N],ma;
int head[N],next[N*2],tov[N*2],tot,zhi[N*2];
int tr[N],al,flag;
bool vis[N],yes;
void build(int x,int y,int z)
{tot++;next[tot]=head[x];tov[tot]=y;head[x]=tot;zhi[tot]=z;}
void dfs(int x) //预遍历
{
vis[x]=true;
int u=head[x],v=tov[u];
while(u){
if(!vis[v])
{
dp[v]=dp[x]+1;dfs(v);
wor[v][0]=zhi[u];
fa[v][0]=x;
}
u=next[u],v=tov[u];
}
}
void LCA(int k,int x,int y)//求lca
{
if(dp[x]<dp[y])swap(x,y);
int dis=dp[x]-dp[y],hh=0;
for(int i=19;i>=0;i--)
if(dis&(1<<i)){hh+=wor[x][i];x=fa[x][i];}
if(x==y)
{
plan[k][0]=x;plan[k][3]=hh;
return;
}
for(int i=19;i>=0;i--)
if(fa[x][i]!=fa[y][i])
{hh+=wor[x][i]+wor[y][i];x=fa[x][i];y=fa[y][i];}
plan[k][0]=fa[x][0];
plan[k][3]=hh+wor[x][0]+wor[y][0];
}
void pre() //预main
{
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&a,&b,&c);
build(a,b,c);build(b,a,c);
}
dp[1]=1;dfs(1);
for(int i=1;i<20;i++)
for(int j=1;j<=n;j++)
{fa[j][i]=fa[fa[j][i-1]][i-1];wor[j][i]=wor[fa[j][i-1]][i-1]+wor[j][i-1];}
for(int i=1;i<=m;i++) //处理出所有线路的相关量
{
scanf("%d%d",&plan[i][1],&plan[i][2]);
LCA(i,plan[i][1],plan[i][2]);
ma=max(ma,plan[i][3]);
}
}
void dfs2(int x,int fa) //check
{
if(yes)return;
int u=head[x],v=tov[u];
while(u){
if(v!=fa)
{
dfs2(v,x);
tr[x]+=tr[v];
}
u=next[u],v=tov[u];
}
if(tr[x]==al&&wor[x][0]>=flag)yes=true;
}
bool check(int x)
{
memset(tr,0,sizeof(tr));al=0;yes=false;
for(int i=1;i<=m;i++)
if(plan[i][3]>x)
{al++;tr[plan[i][1]]++;tr[plan[i][2]]++;tr[plan[i][0]]-=2;}
flag=ma-x;dfs2(1,1);
return yes;
}
void worrk()
{
int le=0,ri=ma,mid=ma;
while(le<ri)
{
mid=(le+ri)>>1;
if(check(mid))ri=mid;
else le=mid+1;
}
if(le==ri)mid=le;
printf("%d\n",mid);
}
int main()
{
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
pre();
worrk();
}