洛谷 P4149 [IOI2011]Race-树分治(点分治,不容斥版)+读入挂-树上求一条路径,权值和等于 K,且边的数量最小

P4149 [IOI2011]Race

题目描述

给一棵树,每条边有权。求一条简单路径,权值和等于 KK,且边的数量最小。

输入格式

第一行包含两个整数 n, Kn,K。

接下来 n - 1n1 行,每行包含三个整数,表示一条无向边的两端和权值。

注意点的编号从 00 开始。

输出格式

输出一个整数,表示最小边数量。

如果不存在这样的路径,输出 -11。

输入输出样例

输入 #1
4 3
0 1 1
1 2 2
1 3 4
输出 #1
2

说明/提示

保证 n \leqslant 2 \times 10^5,n2×105K \leqslant 10^6K106。

 

这道题目我是在cf161D的基础上进行操作的,在维护距离的时候,顺带维护一下边数就可以。

具体细节代码写了注释,心塞,初始化写挫了,调了两天。。。

 

学习博客推荐:

点分治&动态点分治小结

点分治详细解析

点分治&&动态点分治学习笔记

【算法学习】点分治的两种写法与常见套路总结

 

还有一个,没看,关于求重心的

一种基于错误的寻找重心方法的点分治的复杂度分析

 

代码:

  1 //点分治
  2 #include<bits/stdc++.h>
  3 using namespace std;
  4 typedef long long ll;
  5 #pragma GCC optimize(2)
  6 //#define FI(n) FastIO::read(n)
  7 const int inf=0x3f3f3f3f;
  8 const int maxn=2e5+10;
  9 const int maxm=1e6+10;
 10 
 11 int head[maxn<<1],tot;
 12 int root,allnode,n,m,k;
 13 bool vis[maxn];
 14 int deep[maxn],dis[maxn],siz[maxn],maxv[maxn];//deep为维护边数,dis为距离
 15 int minline[maxm],ret[maxn],num[maxn],tmp[maxn];//minline维护对应距离最小的边数,ret保存距离,num保存边数,tmp保存出现过的距离,初始化minline用到
 16 int ans=inf;
 17 
 18 namespace IO{//读入挂
 19     char buf[1<<15],*S,*T;
 20     inline char gc(){
 21         if (S==T){
 22             T=(S=buf)+fread(buf,1,1<<15,stdin);
 23             if (S==T)return EOF;
 24         }
 25         return *S++;
 26     }
 27     inline int read(){
 28         int x; bool f; char c;
 29         for(f=0;(c=gc())<'0'||c>'9';f=c=='-');
 30         for(x=c^'0';(c=gc())>='0'&&c<='9';x=(x<<3)+(x<<1)+(c^'0'));
 31         return f?-x:x;
 32     }
 33     inline long long readll(){
 34         long long x;bool f;char c;
 35         for(f=0;(c=gc())<'0'||c>'9';f=c=='-');
 36         for(x=c^'0';(c=gc())>='0'&&c<='9';x=(x<<3)+(x<<1)+(c^'0'));
 37         return f?-x:x;
 38     }
 39 }
 40 using IO::read;
 41 using IO::readll;
 42 
 43 struct node{
 44     int to,next,val;
 45 }edge[maxn<<1];
 46 
 47 void add(int u,int v,int w)//前向星存图
 48 {
 49     edge[tot].to=v;
 50     edge[tot].next=head[u];
 51     edge[tot].val=w;
 52     head[u]=tot++;
 53 }
 54 
 55 void init()//初始化
 56 {
 57     memset(head,-1,sizeof head);
 58     memset(vis,false,sizeof vis);
 59     for(int i=0;i<=k;i++){//初始化保存最小边数
 60         minline[i]=inf;
 61     }
 62     tot=0;
 63 }
 64 
 65 void get_root(int u,int father)//求重心
 66 {
 67     siz[u]=1;maxv[u]=0;
 68     for(int i=head[u];~i;i=edge[i].next){
 69         int v=edge[i].to;
 70         if(v==father||vis[v]) continue;
 71         get_root(v,u);
 72         siz[u]+=siz[v];
 73         maxv[u]=max(maxv[u],siz[v]);
 74     }
 75 
 76     maxv[u]=max(maxv[u],allnode-siz[u]);
 77     if(maxv[u]<maxv[root]) root=u;
 78 }
 79 
 80 void get_dis(int u,int father)//求距离,维护边数
 81 {
 82     if(dis[u]>k) return ;//超过K,剪枝
 83     ans=min(ans,minline[k-dis[u]]+deep[u]);//更新答案
 84     ret[++ret[0]]=dis[u];num[ret[0]]=deep[u];//保存新得到的数据,距离和边数
 85 
 86     for(int i=head[u];~i;i=edge[i].next){
 87         int v=edge[i].to;
 88         if(v==father||vis[v]) continue;
 89         int w=edge[i].val;
 90         dis[v]=dis[u]+w;//更新距离
 91         deep[v]=deep[u]+1;//更新边数
 92         get_dis(v,u);
 93     }
 94 }
 95 
 96 void cal(int u)
 97 {
 98     int cnt=0;
 99     for(int i=head[u];~i;i=edge[i].next){
100         int v=edge[i].to;
101         int w=edge[i].val;
102         if(vis[v]) continue;
103         ret[0]=0;dis[v]=w;deep[v]=1;//初始化
104         get_dis(v,u);
105 
106         for(int j=1;j<=ret[0];j++){
107            tmp[++cnt]=ret[j];//临时保存出现过的距离
108            minline[ret[j]]=min(minline[ret[j]],num[j]);//更新距离的最小边数
109         }
110     }
111     for(int i=1;i<=cnt;i++){
112         minline[tmp[i]]=inf;//初始化
113     }
114 }
115 
116 void solve(int u)
117 {
118     minline[0]=0;
119     cal(u);
120     vis[u]=1;
121 
122     for(int i=head[u];~i;i=edge[i].next){
123         int v=edge[i].to;
124         if(vis[v]) continue;
125         allnode=siz[v];
126         root=0;maxv[0]=inf;
127         get_root(v,u);
128         solve(root);
129     }
130 }
131 
132 int main()
133 {
134     n=read();k=read();
135     init();
136     for(int i=1;i<n;i++){
137         int u=read(),v=read(),w=read();
138         u++,v++;
139         add(u,v,w);
140         add(v,u,w);
141     }
142     root=0;allnode=n;maxv[0]=inf;
143     get_root(1,0);
144     solve(root);
145     if(ans==inf) printf("-1\n");
146     else printf("%d\n",ans);
147     return 0;
148 }

 

posted @ 2019-10-16 20:18  ZERO-  阅读(377)  评论(0编辑  收藏  举报