码迷,mamicode.com
首页 > 其他好文 > 详细

[coci2015-2016 coii] torrent【树形dp 二分】

时间:2017-03-31 21:56:41      阅读:250      评论:0      收藏:0      [点我收藏+]

标签:main   min   表示   freopen   stdin   out   while   can   学校   

传送门:http://www.hsin.hr/coci/archive/2015_2016/

进去之后点最下面那个。

这道题没有想出来,可惜了,其实不难的。

题目是两个“源”的,我们先考虑单源的问题。先把这个源拉成树根,然后设f(i)为以结点i为树根的子树,全部收到文件所需要的时间,由于同一时间,结点i只能向其中一个子结点传送文件,那么假设son(i, j)表示结点i的第j个子结点(程序中不需要这个数组,这里只是为了叙述方便),那么f(i) = max{ j + f( son(i, j) )}。为了让这个max值最小,很显然需要把f( son(i, j) )按照从大到小排,即子树所需时间越长的,结点i就应该越早传文件给它。这样子就解决了单源的问题。

那么再考虑双源的。考虑从a到b的路径,若a到b的路径上,存在相邻的c,d两个点,很显然,在最佳方案里一定会有文件从a传到c,从b传到d,所以,如果把(c, d)这条边断开也不会影响到答案,这就变成了求两次单源的问题。因此,若把a到b的路径上的边按离a从远到近编号,设f_a(i)表示断开边i后,a所属的那棵树所需要的时间,设f_b(i)表示断开边i后,b所属的那棵树所需要的时间,那么最终答案ans =  min{ max{f_a(i), f_b(i)} }。

这样会tle(我会说std有一个点,在我们学校的垃圾电脑上跑了92s吗?时限可是2s啊喂!你这标程怎么写的!),因此需要优化。很显然(又是显然),断开的边i离a越远,f_a(i)就会越大(其实也可以取等),f_b(i)就会越小(其实也可以取等),因此可以再二分一下断哪条边,就ok了!复杂度是O(n * (logn)^2)

p.s.主函数的前三行是开栈,以免递归爆栈。

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

const int maxn = 300005;

int n, a, b, t1, t2;
int head[maxn], to[maxn << 1], next[maxn << 1], lb, fa[maxn];
char mark[maxn];
std::vector<int> fson[maxn];
struct edge {
	int u, v;
} e[maxn];
int idx;

inline void ist(int aa, int ss) {
	to[lb] = ss;
	next[lb] = head[aa];
	head[aa] = lb;
	++lb;
}
void dfs(int r, int p) {
	fa[r] = p;
	for (int j = head[r]; j != -1; j = next[j]) {
		if (to[j] != p) {
			dfs(to[j], r);
		}
	}
}
int get_ans(int r, int p, int mar) {
	fson[r].clear();
	int son_num = 0, mx = 0;
	for (int j = head[r]; j != -1; j = next[j]) {
		if (to[j] != p && mark[to[j]] != -mar) {
			fson[r].push_back(get_ans(to[j], r, mar));
			++son_num;
		}
	}
	std::sort(fson[r].begin(), fson[r].end());
	for (std::vector<int>::iterator it = fson[r].begin(); it != fson[r].end(); ++it) {
		mx = std::max(mx, *it + son_num);
		--son_num;
	}
	return mx;
}

int main(void) {
	int size = 256 << 20; // 256MB  
	char *p = (char*)malloc(size) + size;  
	__asm__("movl %0, %%esp\n" :: "r"(p)); 
	freopen("torrent.in", "r", stdin);
	freopen("torrent.out", "w", stdout);
	memset(head, -1, sizeof head);
	memset(next, -1, sizeof next);
	scanf("%d%d%d", &n, &a, &b);
	for (int i = 1; i < n; ++i) {
		scanf("%d%d", &t1, &t2);
		ist(t1, t2);
		ist(t2, t1);
	}
	
	dfs(a, 0);
	for (int i = b; i != a; i = fa[i]) {
		e[idx++] = (edge){fa[i], i};
	}
	
	int left = 0, right = idx - 1, mid;
	while (left < right) {
		mid = (left + right) >> 1;
		mark[e[mid].u] = -1;
		mark[e[mid].v] = 1;
		if (get_ans(a, 0, -1) > get_ans(b, 0, 1)) {
			left = mid + 1;
		}
		else {
			right = mid;
		}
		mark[e[mid].u] = mark[e[mid].v] = 0;
	}
	if (!left) {
		mark[e[0].u] = -1;
		mark[e[0].v] = 1;
		printf("%d\n", get_ans(b, 0, 1));
	}
	else {
		int ans1 = -666, ans2 = -666;
		mark[e[left].u] = -1;
		mark[e[left].v] = 1;
		ans1 = get_ans(b, 0, 1);
		mark[e[left].u] = mark[e[left].v] = 0;
		
		mark[e[left - 1].u] = -1;
		mark[e[left - 1].v] = 1;
		ans2 = get_ans(a, 0, -1);
		printf("%d\n", std::min(ans1, ans2));
	}
	return 0;
}

  

[coci2015-2016 coii] torrent【树形dp 二分】

标签:main   min   表示   freopen   stdin   out   while   can   学校   

原文地址:http://www.cnblogs.com/ciao-sora/p/6653426.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!