[JSOI2015] salesman
Posted ztlztl
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[JSOI2015] salesman相关的知识,希望对你有一定的参考价值。
题面
题解
考虑树形 DP , 设 (f[i]) 为 (i) 节点为根的子树最大收益是多少, (h[i]) 代表 (i) 节点的最优方案是否唯一
转移的话拿个堆记一下子节点中 (>0) 的那些, 然后 (h) 跟他们的与一下
若是剩下来的有 (f = 0) 或是跟你选的是一样的, 这个点 (i) 的 (h) 就计为 (0)
Code
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#define mp(i, j) make_pair(i, j)
const int N = 1e5 +5;
const int INF = 0x3f3f3f3f;
using namespace std;
int n, f[N], tm[N], h[N], head[N], cnte;
struct edge { int to, nxt; } e[N << 1];
priority_queue<pair<int, int> > q;
template < typename T >
inline T read()
{
T x = 0, w = 1; char c = getchar();
while(c < '0' || c > '9') { if(c == '-') w = -1; c = getchar(); }
while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * w;
}
inline void adde(int u, int v) { e[++cnte] = (edge) { v, head[u] }, head[u] = cnte; }
void dfs(int u, int fa)
{
for(int v, i = head[u]; i; i = e[i].nxt)
{
v = e[i].to; if(v == fa) continue;
dfs(v, u);
}
while(!q.empty()) q.pop();
int tmp = INF;
for(int v, i = head[u]; i; i = e[i].nxt)
if((v = e[i].to) != fa) q.push(mp(f[v], v));
while(!q.empty() && tm[u] && q.top().first > 0)
{
f[u] += (tmp = q.top().first), h[u] &= h[q.top().second];
q.pop(), tm[u]--;
}
if(!q.empty() && (q.top().first == tmp || !q.top().first)) h[u] = 0;
}
int main()
{
n = read <int> ();
for(int i = 2; i <= n; i++)
f[i] = read <int> ();
for(int i = 2; i <= n; i++)
tm[i] = read <int> () - 1;
tm[1] = INF, memset(h, 1, sizeof(h));
for(int u, v, i = 1; i < n; i++)
u = read <int> (), v = read <int> (), adde(u, v), adde(v, u);
dfs(1, 0);
printf("%d
", f[1]);
puts(h[1] ? "solution is unique" : "solution is not unique");
return 0;
}
以上是关于[JSOI2015] salesman的主要内容,如果未能解决你的问题,请参考以下文章
题解 bzoj4472: [Jsoi2015]salesman (动态规划)