BZ2599（树分治填坑）

第一次写树分治.

刚开始被属于同一分支的点通过根组成的链坑了一发.

然后又被数组下标从零开始坑了一发.

最后数组大小多写了个0又MLE了一发.

感觉有点难看将就一下吧.

#include <queue>
#include <cstdio>
using namespace std;
const int N=200050,M=1000050;
int n,K;
int gfa,rt,tot,ans=1<<30;
int v[M],e[M];
int d[N],dis[N],dep[N],mxs[N],siz[N];
bool vis[N];
struct edge {  
    int u,v,w;  
    edge *nxt;  
    edge (int u,int v,int w,edge *pre): u(u),v(v),w(w){nxt=pre;}  
};  
edge *he[N*2];  
queue <int> q;
void grt(int u,int fa) {
	mxs[u]=0; siz[u]=1;
	for (edge *i=he[u];i;i=i->nxt) if (i->v!=fa && !vis[i->v]) {
		grt(i->v,u); 
		siz[u]+=siz[i->v];
		mxs[u]=max(mxs[u],siz[i->v]);
	}
	mxs[u]=max(mxs[u],siz[gfa]-siz[u]);
	if (mxs[u]<mxs[rt]) rt=u;
}
void DFS(int u,int fa) {
	d[++tot]=u; 
	for (edge *i=he[u];i;i=i->nxt) if (i->v!=fa && !vis[i->v]) {
		dis[i->v]=dis[u]+i->w; dep[i->v]=dep[u]+1; DFS(i->v,u);
	}
}
void dfz(int u) {
	dis[u]=0; dep[u]=0; 
	for (edge *i=he[u];i;i=i->nxt) if (!vis[i->v]) {
		tot=0; v[0]=u; gfa=u; dis[i->v]=dis[u]+i->w; dep[i->v]=dep[u]+1; DFS(i->v,0);
		for (int j=1;j<=tot;j++) if (K-dis[d[j]]>=0) 
			if (v[K-dis[d[j]]]==gfa) ans=min(ans,e[K-dis[d[j]]]+dep[d[j]]);
		for (int j=1;j<=tot;j++) if (dis[d[j]]<K)
			if (v[dis[d[j]]]==gfa) e[dis[d[j]]]=min(dep[d[j]],e[dis[d[j]]]);
			else v[dis[d[j]]]=gfa, e[dis[d[j]]]=dep[d[j]];
	}
	for (edge *i=he[u];i;i=i->nxt) if (!vis[i->v]) {
		gfa=u; rt=0; grt(i->v,0); vis[rt]=1; if (siz[rt]>1) dfz(rt);
	}
}
int main() {
	int i,a,b,c;
	scanf("%d%d",&n,&K); siz[0]=mxs[0]=n;
	for (i=1;i<n;i++) {
		scanf("%d %d %d",&a,&b,&c); a++; b++;
		he[a]=new edge(a,b,c,he[a]);  
        he[b]=new edge(b,a,c,he[b]);  
	}
	gfa=0; rt=0; grt(1,0); vis[rt]=1; dfz(rt);
	if (ans==1<<30) ans=-1;
	printf("%d",ans);
}