首页 > 代码库 > BZOJ 2599 Race
BZOJ 2599 Race
点分写挂调一上午。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<set> #define maxv 200050 #define maxe 400050 #define maxk 1000050 #define inf 1000000007 using namespace std; int n,k,x,y,w,g[maxv],nume=1,ans=inf,lab[maxk],root,sum; int dis[maxv],size[maxv],dep[maxv],mx[maxv],s1[maxv],s2[maxv],top=0; bool vis[maxv]; struct edge { int v,w,nxt; }e[maxe]; int read() { char ch;int data=http://www.mamicode.com/0; while (ch<‘0‘ || ch>‘9‘) ch=getchar(); while (ch>=‘0‘ && ch<=‘9‘) { data=data*10+ch-‘0‘; ch=getchar(); } return data; } void addedge(int u,int v,int w) { e[++nume].v=v;e[nume].w=w; e[nume].nxt=g[u]; g[u]=nume; } void get_root(int x,int fath) { size[x]=1;mx[x]=0; for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (!vis[v] && v!=fath) { get_root(v,x); size[x]+=size[v];mx[x]=max(mx[x],size[v]); } } mx[x]=max(mx[x],sum-size[x]); if (mx[x]<mx[root]) root=x; } void dfs1(int x,int fath) { if (dis[x]<=k) {s1[++top]=dis[x];s2[top]=dep[x];} for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (vis[v] || v==fath) continue; dis[v]=dis[x]+e[i].w;dep[v]=dep[x]+1; dfs1(v,x); } } void dfs2(int x,int fath) { if (dis[x]<=k) lab[dis[x]]=inf; for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (vis[v] || v==fath) continue; dis[v]=dis[x]+e[i].w;dep[v]=dep[x]+1; dfs2(v,x); } } void solve(int x) { vis[x]=true; for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (vis[v]) continue; dis[v]=e[i].w;dep[v]=1; top=0;dfs1(v,0); for (int j=1;j<=top;j++) ans=min(ans,s2[j]+lab[k-s1[j]]); for (int j=0;j<=top;j++) lab[s1[j]]=min(lab[s1[j]],s2[j]); } for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v; if (vis[v]) continue; dis[v]=e[i].w;dep[v]=1; dfs2(v,0); } for (int i=g[x];i;i=e[i].nxt) { int v=e[i].v;if (vis[v]) continue; sum=size[v];root=0;get_root(v,0); solve(root); } } int main() { n=read();k=read(); for (int i=1;i<=k;i++) lab[i]=inf;mx[0]=inf; for (int i=1;i<=n-1;i++) { x=read();y=read();w=read();x++;y++; addedge(x,y,w);addedge(y,x,w); } sum=n;root=0;get_root(1,0); solve(root); if (ans==inf) printf("-1\n"); else printf("%d\n",ans); return 0; }
BZOJ 2599 Race
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。