#基环树，单调队列#洛谷 4381 BZOJ 1791 JZOJ 2304 Island

题目

---

分析

这道题也就是求各棵基环树的直径之和，那么可能在去掉环的某棵子树里，或是经过环，其两端分别在去掉环上所有边之后的两棵不同子树中。首先在基环树中断环求最大的树的直径，就可以解决第一个情况，其次，可以用单调队列实现第二种情况，把环上的点复制一份做dp，就可以解决问题

---

代码

#include <cstdio>
#include <cctype>
#include <queue>
#include <cstring>
#define rr register
using namespace std;
const int N=1000101; typedef long long ll;
struct node{int y,w,next;}e[N<<1]; bool v[N];
ll dis[N<<1],mdis[N],len[N],f[N],lslen,ans;
int k=1,n,ls[N],deg[N],q[N],Q[N<<1],tail;
inline signed iut(){
	rr int ans=0; rr char c=getchar();
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
	return ans;
}
inline void add(int x,int y,int w){e[++k]=(node){y,w,ls[x]}; ls[x]=k;}
inline ll max(ll a,ll b){return a>b?a:b;}
inline void init(){
	rr queue<int>q;
	for (rr int i=1;i<=n;++i) if (deg[i]==1) q.push(i);
	while (q.size()){
		rr int x=q.front(); q.pop();
		for (rr int i=ls[x];i;i=e[i].next)
		if (deg[e[i].y]>1){
			--deg[e[i].y];
			if (deg[e[i].y]==1) q.push(e[i].y);
		}
	} 
}
inline void dp(int x,int fa,int rt){
	for (rr int i=ls[x];i;i=e[i].next)
	if (e[i].y!=fa&&deg[e[i].y]==1){
		dis[e[i].y]=dis[x]+e[i].w;
		mdis[rt]=max(mdis[rt],dis[e[i].y]);
		dp(e[i].y,x,rt);
		len[rt]=max(len[rt],f[x]+f[e[i].y]+e[i].w);
		f[x]=max(f[x],f[e[i].y]+e[i].w);
	}
}
inline void getf(){
	rr bool flag;
	do{
		v[q[tail]]=1,flag=0;
		for (rr int i=ls[q[tail]];i;i=e[i].next)
		if (deg[e[i].y]>1&&!v[e[i].y]){
			flag=1,q[++tail]=e[i].y;
			dis[tail]=dis[tail-1]+e[i].w;
			break;
		}else if (deg[e[i].y]>1) lslen=e[i].w;
	}while (flag);
}
signed main(){
	n=iut();
	for (rr int i=1;i<=n;++i){
		rr int x=iut(),z=iut();
		add(i,x,z),add(x,i,z);
		++deg[i],++deg[x];
	}
	init();
	for (rr int i=1;i<=n;++i) if (deg[i]>1) dp(i,0,i);
	for (rr int i=1;i<=n;++i)
	if (!v[i]&&deg[i]>1){
		memset(q,0,sizeof(q));
		memset(dis,0,sizeof(dis));
		q[tail=1]=i; getf();
		for (rr int i=1;i<=tail;++i)
			q[i+tail]=q[i],dis[i+tail]=dis[i]+dis[tail]+lslen;
		rr int l=1,r=0; rr ll sum=0;
		for (rr int i=1;i<=2*tail;++i){
			while (l<=r&&i-Q[l]>=tail) ++l;
			if (l<=r) sum=max(sum,mdis[q[i]]+mdis[q[Q[l]]]+dis[i]-dis[Q[l]]);
			while (l<=r&&mdis[q[i]]>mdis[q[Q[r]]]+dis[i]-dis[Q[r]]) --r;
			Q[++r]=i;
		}
		for (rr int i=1;i<=tail;++i) sum=max(sum,len[q[i]]);
		ans+=sum;
	}
	return !printf("%lld",ans);
}

---

upd：8.17

分析

可以看出数据是一个内向的基环树森林，所以说环的方向是可以确定的，所以这样常数就会大大减少，也不用手工栈了，其它其实差不多

---

代码

#include <bits/stdc++.h>
#define rr register
using namespace std;
const int N=1000101; typedef long long ll;
int w[N],deg[N],y[N],now[N<<1],Q[N<<1],n;
ll ans,dis[N],f[N],dp[N],s[N<<1]; bool v[N];
inline signed iut(){
	rr int ans=0; rr char c=getchar();
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
	return ans;
}
inline signed max(int a,int b){return a>b?a:b;}
signed main(){
	n=iut(); rr queue<int>q;
	for (rr int i=1;i<=n;++i)
		++deg[y[i]=iut()],w[i]=iut();
	for (rr int i=1;i<=n;++i)
	    if (!deg[i]) q.push(i);
	while (q.size()){
		rr int x=q.front(); q.pop(); 
		if (!(--deg[y[x]])) q.push(y[x]);
		f[y[x]]=max(f[y[x]],dis[y[x]]+dis[x]+w[x]);
		dis[y[x]]=max(dis[y[x]],dis[x]+w[x]);
		dp[y[x]]=max(dp[y[x]],max(dp[x],f[y[x]]));
    }
    for (rr int i=1;i<=n;++i)
    if (deg[i]&&!v[i]){
    	rr int tail=0,l=1,r=0; rr ll t=0;
    	for (rr int j=i;!v[j];j=y[j])
    		v[now[++tail]=j]=1,t=max(t,dp[j]);
		for (rr int j=1;j<=tail;++j) now[j+tail]=now[j]; tail<<=1;
		for (rr int j=1;j<=tail;++j){
			s[j]=s[j-1]+w[now[j-1]];
			while (l<=r&&j-Q[l]>=(tail>>1)) ++l;
			if (l<=r) t=max(t,dis[now[Q[l]]]-s[Q[l]]+dis[now[j]]+s[j]);
			while (l<=r&&dis[now[Q[r]]]-s[Q[r]]<dis[now[j]]-s[j]) --r; Q[++r]=j;
		}
		ans+=t;
	}
	return !printf("%lld",ans);
}