[coci2015-2016 coii] torrent树形dp 二分
Posted ciao_sora
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[coci2015-2016 coii] torrent树形dp 二分相关的知识,希望对你有一定的参考价值。
传送门: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 二分的主要内容,如果未能解决你的问题,请参考以下文章
[COCI]coci2015/2016 nekameleoni