博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
poj - 4045 - Power Station
阅读量:6483 次
发布时间:2019-06-23

本文共 1931 字,大约阅读时间需要 6 分钟。

题意:一棵有n个结点的树,要取其中的一个结点,使得该结点到其他所有结点的距离和dis最小,即损耗I * I * R * dis最小,输出最小损耗和该结点(有多个的话按结点编号从小到大输出)(3 <= n <= 50000, 1 <= I <= 10, 1 <= R <= 50)。

题目链接:

——>>怒刷树状dp。。。

设cnt[i]为以i为根的子树的结点数,d[i]为以i为根的子树中所有结点到i的距离和,一次dfs求出*cnt和*d,则

状态转移方程为:cnt[x] += cnt[v[e]];

d[x] += d[v[e]] + cnt[v[e]];(v[e]为x的子结点)

设f[i]为以i为根,刚才dfs中i的父结点为孩子,新生子树中所有结点到i的距离和,再一次dfs,即dp,则

状态转移方程为:f[x] = d[fa] - d[x] - cnt[x] + f[fa] + n -cnt[x] = d[fa] - d[x] - 2 * cnt[x] + f[fa] + n;(x为第一次dfs中fa的子结点)。

注意:用64位整数

 

#include 
#include
#include
using namespace std;const int maxn = 50000 + 10;const int INF = 0x3f3f3f3f;int n, I, R, head[maxn], nxt[maxn<<1], u[maxn<<1], v[maxn<<1], ecnt, cnt[maxn];long long d[maxn], f[maxn];void init(){ ecnt = 0; memset(head, -1, sizeof(head));}void addEdge(int uu, int vv){ u[ecnt] = uu; v[ecnt] = vv; nxt[ecnt] = head[uu]; head[uu] = ecnt; ecnt++;}void read(){ int i, uu, vv; scanf("%d%d%d", &n, &I, &R); for(i = 0; i < n-1; i++){ scanf("%d%d", &uu, &vv); addEdge(uu, vv); addEdge(vv, uu); }}void dfs(int x, int fa){ d[x] = 0; for(int e = head[x]; e != -1; e = nxt[e]) if(v[e] != fa){ dfs(v[e], x); cnt[x] += cnt[v[e]]; d[x] += d[v[e]] + cnt[v[e]]; }}void dp(int x, int fa){ if(x == 1) f[x] = 0; else f[x] = d[fa] - d[x] - 2 * cnt[x] + f[fa] + n; for(int e = head[x]; e != -1; e = nxt[e]) if(v[e] != fa) dp(v[e], x);}void solve(){ int i, flag = 1; long long Min = INF; for(int i = 1; i <= n; i++) cnt[i] = 1; dfs(1, -1); dp(1, -1); for(i = 1; i <= n; i++) Min = min(Min, d[i] + f[i]); printf("%I64d\n", I * I * R * Min); for(i = 1; i <= n; i++) if(d[i] + f[i] == Min) { if(flag) flag = 0; else putchar(' '); printf("%d", i); } puts("\n");}int main(){ int T; scanf("%d", &T); while(T--){ init(); read(); solve(); } return 0;}

 

 

转载地址:http://swbuo.baihongyu.com/

你可能感兴趣的文章
C/C++中extern关键字详解
查看>>
Eclipse 最有用的快捷键
查看>>
K & DN 的前世今生(微软开源命名变革)
查看>>
--@angularJS--angular与BootStrap3的应用
查看>>
u3d单词学习plane
查看>>
10款很好用的 jQuery 图片滚动插件
查看>>
Flask服务入门案例
查看>>
ReadWriteLock与ReentrantReadWriteLock
查看>>
Atitit.软件命名空间 包的命名统计 及命名表(2000个名称) 方案java package...
查看>>
新手指导:教你如何查看识别hadoop是32位还是64位
查看>>
Codeforces Round #180 (Div. 2) D. Fish Weight 贪心
查看>>
Gradle sourceCompatibility has no effect to subprojects(转)
查看>>
百度指数分析
查看>>
使用Mkdocs构建你的项目文档
查看>>
三分钟读懂TT猫分布式、微服务和集群之路
查看>>
fn project 运行时配置选项
查看>>
你的leader还在考核你的千行代码Bug率吗?
查看>>
多块盘制作成一个lvm
查看>>
InnoDB多版本
查看>>
贪心算法 - 活动选择问题
查看>>