BZ2599（树分治填坑）

来源：互联网发布：网络暴力现象分析编辑：程序博客网时间：2024/05/16 08:56

第一次写树分治.

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

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

最后数组大小多写了个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);}

0 0