bzoj 2599: [IOI2011]Race（树的点分治）

最新推荐文章于 2019-08-22 21:43:48 发布

Jaihk662

最新推荐文章于 2019-08-22 21:43:48 发布

阅读量450

点赞数 1

分类专栏： # 贪心、动规与分治文章标签： bzoj

本文链接：https://blog.csdn.net/jaihk662/article/details/78161182

版权

贪心、动规与分治专栏收录该内容

118 篇文章 1 订阅

订阅专栏

2599: [IOI2011]Race

Time Limit: 70 Sec Memory Limit: 128 MB
Submit: 3862 Solved: 1144
[ Submit][ Status][ Discuss]

Description

给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N <= 200000, K <= 1000000

Input

第一行两个整数 n, k
第二..n行每行三个整数表示一条无向边的两端和权值 (注意点的编号从0开始)

Output

一个整数表示最小边数量如果不存在这样的路径输出-1

Sample Input

4 3

0 1 1

1 2 2

1 3 4

Sample Output

http://blog.csdn.net/jaihk662/article/details/77429708

每次拉出重心，然后爆搜所有经过该节点的路径，看存不存在等于k的

#include<stdio.h>
#include<vector>
using namespace std;
typedef struct
{
	int y;
	int len;
}Point;
Point now, s[200005];
vector<Point> G[200005];
int ans, cnt, n, k, bet, heart, vis[200005], size[200005], dp[2000005], st[200005];
void Sech1(int u, int p)
{
	int i, v;
	size[u] = 1;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i].y;
		if(v==p || vis[v])
			continue;
		Sech1(v, u);
		size[u] += size[v];
	}
}
void Sech2(int u, int p, int &psiz)
{
	int bsiz, v, i;
	bsiz = psiz-size[u];
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i].y;
		if(v==p || vis[v])
			continue;
		bsiz = max(bsiz, size[v]);
		Sech2(v, u, psiz);
	}
	if(bsiz<bet)
	{
		bet = bsiz;
		heart = u;
	}
}
void Sech(int u, int p, int len, int now)
{
	int i, v;
	if(k-len>0 && dp[k-len]!=0 || k==len)
		ans = min(ans, now+dp[k-len]);
	if(k-len>0)
		s[++cnt].y = now, s[cnt].len = len;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i].y;
		if(vis[v] || v==p)
			continue;
		Sech(v, u, len+G[u][i].len, now+1);
	}
}
void Count(int u)
{
	int i, j, v, all;
	all = 0;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i].y;
		if(vis[v])
			continue;
		cnt = 0;
		Sech(v, u, G[u][i].len, 1);
		for(j=1;j<=cnt;j++)
		{
			if(dp[s[j].len]==0 && s[j].len!=0)
				dp[s[j].len] = s[j].y, st[++all] = s[j].len;
			else
				dp[s[j].len] = min(dp[s[j].len], s[j].y);
		}
	}
	if(dp[k]!=0)
		ans = min(ans, dp[k]);
	for(i=1;i<=all;i++)
		dp[st[i]] = 0;
}
void Divide(int u)
{
	int i, v;
	Sech1(u, 0);
	bet = 200005;
	Sech2(u, 0, size[u]);
	vis[heart] = 1;
	u = heart;
	Count(u);
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i].y;
		if(vis[v])
			continue;
		Divide(v);
	}
}
int main(void)
{
	int x, y, len, i;
	//freopen("in.txt", "r", stdin);
	scanf("%d%d", &n, &k);
	for(i=1;i<=n-1;i++)
	{
		scanf("%d%d%d", &x, &y, &len);
		x++, y++;
		now.y = y, now.len = len;
		G[x].push_back(now);
		now.y = x;
		G[y].push_back(now);
	}
	if(k==0)
	{
		printf("0\n");
		return 0;
	}
	ans = 2345667;
	Divide(1);
	if(ans==2345667)
		ans -= 2345668;
	printf("%d\n", ans);
	return 0;
}