CSTC2018暴力写挂

# 解法一

$dep(x)+dep(y)-dep(lca)-dep'(lca)$

$=\frac{1}{2}(dep(x)+dep(y)-2*dep(lca)+dep(x)+dep(y))-dep'(lca)$

$=\frac{1}{2}(dis(x,y)+dep(x)+dep(y))-dep'(lca)$

#include<set>
#include<map>
#include<queue>
#include<cmath>
#include<stack>
#include<cstdio>
#include<vector>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define INF (1<<30)
#define pr pair<int,ll>
using namespace std;
char *p1,*p2,buf[100000];
int rd() {int x=0,f=1; char c=nc(); while(!isdigit(c)) {if(c=='-') f=-1; c=nc();} while(isdigit(c)) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x*f;}
ll rd2() {ll x=0,f=1; char c=nc(); while(!isdigit(c)) {if(c=='-') f=-1; c=nc();} while(isdigit(c)) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x*f;}
int n,m;
int x,y;
ll ans;
ll z;
int cnt;
int col[400010];
struct miku
{
int x;
ll val;
}t[400010];
namespace virtual_tree
{
ll res;
int tot;
int top;
int dfn=0;
ll mid_edge;
ll d[400010];
ll v[400010];
int s[400010];
ll val[800010];
int st[400010];
int to[800010];
int lg[800010];
int vis[400010];
int dep[400010];
int next[800010];
int dp[400010][2];
int f[800010][20];
vector<int>q[400010];
inline void add(int x,int y,ll z)
{
to[tot]=y;
val[tot]=z;
}
inline bool cmp(const miku &a,const miku &b)
{
return s[a.x]<s[b.x];
}
void dfs(int x,int fa)
{
f[++dfn][0]=x;
s[x]=dfn;
{
if(to[i]!=fa)
{
dep[to[i]]=dep[x]+1;
d[to[i]]=d[x]+val[i];
dfs(to[i],x);
f[++dfn][0]=x;
}
}
}
inline int mn(int x,int y)
{
return dep[x]<dep[y]?x:y;
}
inline void ST()
{
for(int i=2;i<=dfn;i++)
{
lg[i]=lg[i>>1]+1;
}
for(int j=1;j<=19;j++)
{
for(int i=1;i+(1<<j)-1<=dfn;i++)
{
f[i][j]=mn(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}
}
}
inline int lca(int x,int y)
{
x=s[x],y=s[y];
if(x>y)
{
swap(x,y);
}
int len=lg[y-x+1];
return mn(f[x][len],f[y-(1<<len)+1][len]);
}
inline void insert(int x)
{
int fa=lca(x,st[top]);
if(!vis[fa])
{
vis[fa]=1;
v[fa]=-1ll<<60;
dp[fa][0]=dp[fa][1]=0;
}
while(top>1&&dep[st[top-1]]>=dep[fa])
{
q[st[top-1]].push_back(st[top]);
top--;
}
if(fa!=st[top])
{
q[fa].push_back(st[top]);
st[top]=fa;
}
st[++top]=x;
}
inline int merge(int x,int y)
{
if(!x||!y)
{
return x+y;
}
return v[x]>v[y]?x:y;
}
inline void query(int x,int y)
{
if(!x||!y)
{
return ;
}
res=max(res,v[x]+v[y]);
}
void tree_dp(int x)
{
int len=q[x].size();
for(int i=0;i<len;i++)
{
int to=q[x][i];
tree_dp(to);
res=-1ll<<60;
query(dp[x][0],dp[to][1]);
query(dp[x][1],dp[to][0]);
ans=max(ans,res+mid_edge-2*d[x]);
dp[x][0]=merge(dp[x][0],dp[to][0]);
dp[x][1]=merge(dp[x][1],dp[to][1]);
}
vis[x]=0;
q[x].clear();
}
inline void build(ll value)
{
mid_edge=value;
for(int i=1;i<=cnt;i++)
{
vis[t[i].x]=1;
v[t[i].x]=t[i].val;
dp[t[i].x][col[t[i].x]-1]=t[i].x;
dp[t[i].x][(col[t[i].x]-1)^1]=0;
col[t[i].x]=0;
}
sort(t+1,t+1+cnt,cmp);
top=0;
if(t[1].x!=1)
{
st[++top]=1;
}
for(int i=1;i<=cnt;i++)
{
insert(t[i].x);
}
while(top>1)
{
q[st[top-1]].push_back(st[top]);
top--;
}
tree_dp(1);
}
inline void work()
{
dfs(1,0);
ST();
}
};
namespace edge_partation
{
int tot;
int num;
int root;
ll d[800010];
ll val[1600010];
int to[1600010];
int vis[800010];
int size[800010];
int next[1600010];
vector<pr>v[400010];
inline void push(int x,int y,ll z)
{
v[x].push_back(make_pair(y,z));
}
inline void add(int x,int y,ll z)
{
to[tot]=y;
val[tot]=z;
}
void rebuild(int x,int fa)
{
int tmp=0;
int last=0;
int len=v[x].size();
for(int i=0;i<len;i++)
{
int to=v[x][i].first;
int val=v[x][i].second;
if(to==fa)
{
continue;
}
tmp++;
if(tmp==1)
{
last=x;
}
else if(tmp==len-(x!=1))
{
}
else
{
m++;
last=m;
}
}
for(int i=0;i<len;i++)
{
if(v[x][i].first==fa)
{
continue;
}
rebuild(v[x][i].first,x);
}
}
void dfs(int x,int fa)
{
{
if(to[i]!=fa)
{
d[to[i]]=d[x]+val[i];
dfs(to[i],x);
}
}
}
void getroot(int x,int fa,int sum)
{
size[x]=1;
{
if(!vis[i>>1]&&to[i]!=fa)
{
getroot(to[i],x,sum);
size[x]+=size[to[i]];
int mx_size=max(size[to[i]],sum-size[to[i]]);
if(mx_size<num)
{
num=mx_size;
root=i;
}
}
}
}
void dfs2(int x,int fa,ll dep,int opt)
{
if(x<=n)
{
col[x]=opt;
t[++cnt]=(miku){x,d[x]+dep};
}
{
if(!vis[i>>1]&&to[i]!=fa)
{
dfs2(to[i],x,dep+val[i],opt);
}
}
}
void partation(int x,int sum)
{
num=INF;
getroot(x,0,sum);
if(num==INF)
{
return ;
}
int now=root;
vis[now>>1]=1;
cnt=0;
dfs2(to[now],0,0ll,1);
dfs2(to[now^1],0,0ll,2);
virtual_tree::build(val[now]);
int sz=size[to[now]];
partation(to[now],sz);
partation(to[now^1],sum-sz);
}
inline void work()
{
tot=1;
rebuild(1,0);
dfs(1,0);
partation(1,m);
}
};
int main()
{
m=n=rd();
ans=-1ll<<60;
for(int i=1;i<n;i++)
{
x=rd(),y=rd(),z=rd2();
edge_partation::push(x,y,z);
edge_partation::push(y,x,z);
}
for(int i=1;i<n;i++)
{
x=rd(),y=rd(),z=rd2();
}
virtual_tree::work();
edge_partation::work();
ans>>=1;
for(int i=1;i<=n;i++)
{
ans=max(ans,edge_partation::d[i]-virtual_tree::d[i]);
}
printf("%lld",ans);
}

# 解法二

$dep(x)+dep(y)-dep(lca)-dep'(lca)$

$=dep(y)+(dep(x)-dep(lca))-dep'(lca)$(这好像除了加了个括号、换了下位置啥都没变啊QAQ)

#include<set>
#include<map>
#include<queue>
#include<cmath>
#include<stack>
#include<cstdio>
#include<vector>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define INF (1<<30)
#define pr pair<int,ll>
using namespace std;
char *p1,*p2,buf[100000];
int rd() {int x=0,f=1; char c=nc(); while(!isdigit(c)) {if(c=='-') f=-1; c=nc();} while(isdigit(c)) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x*f;}
ll rd2() {ll x=0,f=1; char c=nc(); while(!isdigit(c)) {if(c=='-') f=-1; c=nc();} while(isdigit(c)) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x*f;}
int n,m;
int x,y;
ll ans;
ll z;
int cnt;
int col[400010];
struct miku
{
int x;
ll val;
}t[400010];
namespace virtual_tree
{
ll res;
int tot;
int top;
int dfn=0;
ll d[400010];
ll v[400010];
int s[400010];
ll val[800010];
int st[400010];
int to[800010];
int lg[800010];
int vis[400010];
int dep[400010];
int next[800010];
int dp[400010][2];
int f[800010][20];
vector<int>q[400010];
inline void add(int x,int y,ll z)
{
to[tot]=y;
val[tot]=z;
}
inline bool cmp(const miku &a,const miku &b)
{
return s[a.x]<s[b.x];
}
inline void dfs(int x,int fa)
{
f[++dfn][0]=x;
s[x]=dfn;
{
if(to[i]!=fa)
{
dep[to[i]]=dep[x]+1;
d[to[i]]=d[x]+val[i];
dfs(to[i],x);
f[++dfn][0]=x;
}
}
}
inline int mn(int x,int y)
{
return dep[x]<dep[y]?x:y;
}
inline void ST()
{
for(int i=2;i<=dfn;i++)
{
lg[i]=lg[i>>1]+1;
}
for(int j=1;j<=19;j++)
{
for(int i=1;i+(1<<j)-1<=dfn;i++)
{
f[i][j]=mn(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}
}
}
inline int lca(int x,int y)
{
x=s[x],y=s[y];
if(x>y)
{
swap(x,y);
}
int len=lg[y-x+1];
return mn(f[x][len],f[y-(1<<len)+1][len]);
}
inline void insert(int x)
{
int fa=lca(x,st[top]);
if(!vis[fa])
{
vis[fa]=1;
v[fa]=-1ll<<60;
dp[fa][0]=dp[fa][1]=0;
}
while(top>1&&dep[st[top-1]]>=dep[fa])
{
q[st[top-1]].push_back(st[top]);
top--;
}
if(fa!=st[top])
{
q[fa].push_back(st[top]);
st[top]=fa;
}
st[++top]=x;
}
inline int merge(int x,int y)
{
if(!x||!y)
{
return x+y;
}
return v[x]>v[y]?x:y;
}
inline void query(int x,int y)
{
if(!x||!y)
{
return ;
}
res=max(res,v[x]+v[y]);
}
inline void tree_dp(int x)
{
int len=q[x].size();
for(int i=0;i<len;i++)
{
int to=q[x][i];
tree_dp(to);
res=-1ll<<60;
query(dp[x][0],dp[to][1]);
query(dp[x][1],dp[to][0]);
ans=max(ans,res-d[x]);
dp[x][0]=merge(dp[x][0],dp[to][0]);
dp[x][1]=merge(dp[x][1],dp[to][1]);
}
vis[x]=0;
q[x].clear();
}
inline void build()
{
for(int i=1;i<=cnt;i++)
{
vis[t[i].x]=1;
v[t[i].x]=t[i].val;
dp[t[i].x][col[t[i].x]-1]=t[i].x;
dp[t[i].x][(col[t[i].x]-1)^1]=0;
col[t[i].x]=0;
}
sort(t+1,t+1+cnt,cmp);
top=0;
if(t[1].x!=1)
{
st[++top]=1;
}
for(int i=1;i<=cnt;i++)
{
insert(t[i].x);
}
while(top>1)
{
q[st[top-1]].push_back(st[top]);
top--;
}
tree_dp(1);
}
inline void work()
{
dfs(1,0);
ST();
}
};
namespace edge_partation
{
int tot;
int num;
int root;
int dfn=0;
ll d[800010];
int s[800010];
int lg[1600010];
ll val[1600010];
int to[1600010];
int vis[800010];
int dep[800010];
int size[800010];
int next[1600010];
int f[1600010][21];
vector<pr>v[400010];
inline void push(int x,int y,ll z)
{
v[x].push_back(make_pair(y,z));
}
inline void add(int x,int y,ll z)
{
to[tot]=y;
val[tot]=z;
}
inline void rebuild(int x,int fa)
{
int tmp=0;
int last=0;
int len=v[x].size();
for(int i=0;i<len;i++)
{
int to=v[x][i].first;
int val=v[x][i].second;
if(to==fa)
{
continue;
}
tmp++;
if(tmp==1)
{
last=x;
}
else if(tmp==len-(x!=1))
{
}
else
{
m++;
last=m;
}
}
for(int i=0;i<len;i++)
{
if(v[x][i].first==fa)
{
continue;
}
rebuild(v[x][i].first,x);
}
}
inline void dfs(int x,int fa)
{
f[++dfn][0]=x;
s[x]=dfn;
{
if(to[i]!=fa)
{
dep[to[i]]=dep[x]+1;
d[to[i]]=d[x]+val[i];
dfs(to[i],x);
f[++dfn][0]=x;
}
}
}
inline int mn(int x,int y)
{
return dep[x]<dep[y]?x:y;
}
inline void ST()
{
for(int i=2;i<=dfn;i++)
{
lg[i]=lg[i>>1]+1;
}
for(int j=1;j<=20;j++)
{
for(int i=1;i+(1<<j)-1<=dfn;i++)
{
f[i][j]=mn(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}
}
}
inline int lca(int x,int y)
{
x=s[x],y=s[y];
if(x>y)
{
swap(x,y);
}
int len=lg[y-x+1];
return mn(f[x][len],f[y-(1<<len)+1][len]);
}
inline void getroot(int x,int fa,int sum)
{
size[x]=1;
{
if(!vis[i>>1]&&to[i]!=fa)
{
getroot(to[i],x,sum);
size[x]+=size[to[i]];
int mx_size=max(size[to[i]],sum-size[to[i]]);
if(mx_size<num)
{
num=mx_size;
root=i;
}
}
}
}
inline void dfs2(int x,int fa,int rt,int opt)
{
if(x<=n)
{
col[x]=opt;
ll value=rt?d[x]-d[lca(x,rt)]:d[x];
t[++cnt]=(miku){x,value};
}
{
if(!vis[i>>1]&&to[i]!=fa)
{
dfs2(to[i],x,rt,opt);
}
}
}
inline void partation(int x,int sum)
{
num=INF;
getroot(x,0,sum);
if(num==INF)
{
return ;
}
int now=root;
vis[now>>1]=1;
cnt=0;
dfs2(x,0,to[now],1);
dfs2(to[now],0,0,2);
virtual_tree::build();
int sz=size[to[now]];
partation(to[now],sz);
partation(x,sum-sz);
}
inline void work()
{
tot=1;
rebuild(1,0);
dfs(1,0);
ST();
partation(1,m);
}
};
int main()
{
m=n=rd();
ans=-1ll<<60;
for(int i=1;i<n;i++)
{
x=rd(),y=rd(),z=rd2();
edge_partation::push(x,y,z);
edge_partation::push(y,x,z);
}
for(int i=1;i<n;i++)
{
x=rd(),y=rd(),z=rd2();
}