题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4679

题意:给一棵树,每条边上都有一个权值,去掉树上任意一条边之后,分成两个子树,两个子树的最长路与这条边上的权值相乘的到一个乘积。问去掉那一条边可以使这个乘积最小。

分析:求出树的直径,然后判断边是否树的直径上,如果是的话,ans=w*mx_len(mx_len为树的直径),否则从直径的两个端点出发dfs处理以每个节点为根节点时的直径及根节点到子树中的最长路和次长路。

#pragma comment(linker,"/STACK:102400000,102400000")
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <stack>
#include <vector>
#include <set>
#include <map>
#define LL long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define N 100010
#define FILL(a,b) (memset(a,b,sizeof(a)))
using namespace std; struct edge
{
int v,w,id,next;
edge(){}
edge(int v,int w,int id,int next):v(v),w(w),id(id),next(next){}
}e[*N];
int head[N*],tot,n,mx_len;
int dp[N][];//dp[u][0]表示该节点到子树中的最长路,dp[u][1]表示次长路,dp[u][2]表示这颗树的直径
int path[N];//保存树的直径上的路径
int mark[N];//直径上边的标志
int ans[N],st,ed;//st,ed为树的直径的两个端点
void addedge(int u,int v,int w,int id)
{
e[tot]=edge(v,w,id,head[u]);
head[u]=tot++;
}
void dfs_len(int u,int fa,int len)
{
if(len>=mx_len)mx_len=len,ed=u;
for(int i=head[u];~i;i=e[i].next)
{
int v=e[i].v;
if(v==fa)continue;
path[v]=u;
dfs_len(v,u,len+);
}
}
void dfs(int u,int fa)
{
dp[u][]=dp[u][]=dp[u][]=;
for(int i=head[u];~i;i=e[i].next)
{
int v=e[i].v;
if(v==fa)continue;
dfs(v,u);
int tmp=dp[v][]+;
if(tmp>dp[u][])
{
dp[u][]=dp[u][];
dp[u][]=tmp;
}
else if(tmp>dp[u][])
{
dp[u][]=tmp;
}
dp[u][]=max(dp[u][],dp[v][]);
}
dp[u][]=max(dp[u][],dp[u][]+dp[u][]);
}
void solve(int u,int fa)
{
for(int i=head[u];~i;i=e[i].next)
{
int v=e[i].v,w=e[i].w,id=e[i].id;
if(v==fa)continue;
solve(v,u);
if(mark[u]&&mark[v])ans[id]=max(ans[id],w*dp[v][]);
else ans[id]=w*mx_len;
}
}
int main()
{
int u,v,w,T,cas=;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
FILL(head,-);FILL(mark,);
FILL(ans,);tot=;
for(int i=;i<n;i++)
{
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w,i);
addedge(v,u,w,i);
}
mx_len=;dfs_len(,-,);
st=ed;dfs_len(st,-,);
mark[st]=;path[st]=-;
int tmp=ed;
while(path[tmp]!=-)
{
mark[tmp]=;
tmp=path[tmp];
}
dfs(st,-);solve(st,-);
dfs(ed,-);solve(ed,-);
int ret=;
for(int i=;i<n;i++)
if(ans[i]<ans[ret])ret=i;
printf("Case #%d: %d\n",cas++,ret);
}
}

后来发现直接预处理好每个点到达树的直径的两个端点的距离,就可以判断边是否在直径上了,且如果边在直径上,那么两个子树中的最大值那条路,一定是以整棵树的最长路的两个端点为起始点的。好像使用vector会超时。。。

#pragma comment(linker,"/STACK:102400000,102400000")
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
#define N 100010
#define inf 0x7fffffff int n, s, t, len, id, ans;
int ds[N], dt[N];//记录每个点到树的直径的两个端点的距离
struct node {
int v, w, id,next;
node() {}
node(int _v, int _w, int _id,int _next) : v(_v), w(_w), id(_id) ,next(_next){}
}e[N<<];
int head[N<<],tot;
void addedge(int u,int v,int w,int id)
{
e[tot]=node(v,w,id,head[u]);
head[u]=tot++;
}
void dfs(int now, int fa) {
int u;
for (int i=head[now];~i;i=e[i].next)
if ((u = e[i].v) != fa) {
ds[u] = ds[now] + ;
dfs(u, now);
}
}
void dfs2(int now, int fa) {
int u;
for (int i=head[now];~i;i=e[i].next)
if ((u = e[i].v) != fa) {
dt[u] = dt[now] + ;
dfs2(u, now);
}
}
void work(int now, int fa) {
int u, w;
for(int i=head[now];~i;i=e[i].next)
if ((u = e[i].v) != fa) {
if (ds[now] + + dt[u] == len)
w = e[i].w * max(ds[now], dt[u]);
else w = e[i].w * len;
if (w < ans) { ans = w, id = e[i].id; }
else if (w == ans && e[i].id < id)
id = e[i].id;
work(u, now);
}
}
int main() { int T,a,b,c;
scanf("%d", &T);
for (int cas=; cas<=T; cas++) {
scanf("%d", &n);
memset(head,-,sizeof(head));tot=;
for (int i=; i<n; i++) {
scanf("%d%d%d", &a, &b, &c);
addedge(a,b,c,i);
addedge(b,a,c,i);
}
ds[] = ;
dfs(, );
ds[s=] = ; for (int i=; i<=n; i++) if (ds[i] > ds[s]) s = i; ds[s] = ; dfs(s, );
t = ; for (int i=; i<=n; i++) if (ds[i] > ds[t]) t = i;
len = ds[t]; dt[t] = ;
dfs2(t, ); id = ans = inf;
work(s, );
printf("Case #%d: %d\n", cas, id);
}
return ;
}

hdu4679(树形dp)的更多相关文章

  1. 树形DP 2013多校8(Terrorist’s destroy HDU4679)

    题意: There is a city which is built like a tree.A terrorist wants to destroy the city's roads. But no ...

  2. HDU-4679 Terrorist’s destroy 树形DP,维护

    题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4679 题意:给一颗树,每个边有一个权值,要你去掉一条边权值w剩下的两颗子树中分别的最长链a,b,使得w ...

  3. poj3417 LCA + 树形dp

    Network Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 4478   Accepted: 1292 Descripti ...

  4. COGS 2532. [HZOI 2016]树之美 树形dp

    可以发现这道题的数据范围有些奇怪,为毛n辣么大,而k只有10 我们从树形dp的角度来考虑这个问题. 如果我们设f[x][k]表示与x距离为k的点的数量,那么我们可以O(1)回答一个询问 可是这样的话d ...

  5. 【BZOJ-4726】Sabota? 树形DP

    4726: [POI2017]Sabota? Time Limit: 20 Sec  Memory Limit: 128 MBSec  Special JudgeSubmit: 128  Solved ...

  6. 树形DP+DFS序+树状数组 HDOJ 5293 Tree chain problem(树链问题)

    题目链接 题意: 有n个点的一棵树.其中树上有m条已知的链,每条链有一个权值.从中选出任意个不相交的链使得链的权值和最大. 思路: 树形DP.设dp[i]表示i的子树下的最优权值和,sum[i]表示不 ...

  7. 树形DP

    切题ing!!!!! HDU  2196 Anniversary party 经典树形DP,以前写的太搓了,终于学会简单写法了.... #include <iostream> #inclu ...

  8. BZOJ 2286 消耗战 (虚树+树形DP)

    给出一个n节点的无向树,每条边都有一个边权,给出m个询问,每个询问询问ki个点,问切掉一些边后使得这些顶点无法与顶点1连接.最少的边权和是多少.(n<=250000,sigma(ki)<= ...

  9. POJ2342 树形dp

    原题:http://poj.org/problem?id=2342 树形dp入门题. 我们让dp[i][0]表示第i个人不去,dp[i][1]表示第i个人去 ,根据题意我们可以很容易的得到如下递推公式 ...

  10. hdu1561 The more, The Better (树形dp+背包)

    题目链接:http://acm.split.hdu.edu.cn/showproblem.php?pid=1561 思路:树形dp+01背包 //看注释可以懂 用vector建树更简单. 代码: #i ...

随机推荐

  1. 1.1.2-学习Opencv与MFC混合编程之---画图工具 画直线 画圆 画矩形

    源代码地址:http://download.csdn.net/detail/nuptboyzhb/3961685 画图工具 1.     画直线 Ø  增加‘直线’菜单项,建立类向导: Ø  对CXX ...

  2. uvc摄像头代码解析6

    10.扫描视频设备链和注册视频设备 10.1 uvc视频链 struct uvc_video_chain { //uvc视频链 struct uvc_device *dev; //uvc设备 stru ...

  3. Google App Engine 学习和实践

    这个周末玩了玩Google App Engine,随手写点东西,算是学习笔记吧.不当之处,请多多指正. 作者:liigo,2009/04/26夜,大连 原创链接:http://blog.csdn.ne ...

  4. oracle至mysql该指南的数据模式()任意数据源之间的跨导应用

    为了产生的一些资源的库的释放.需要API模块迁移到mysql在,需要引导数据. 试用oracle to mysql工具.当迁移错误不说,如此大量的数据的,有了这样简陋的工具是不太可靠. 意外的发现工具 ...

  5. Swift - 内存泄露原因(循环强引用)及解决办法

    Swift使用自动引用计数(ARC)来管理应用程序的内存使用.在大多是情况下,并不需要考虑内存的管理.当实例不再需要的时候,ARC会自动释放这些实例所使用的内存. 但ARC并不是绝对安全的.下面两种情 ...

  6. Flexigrid去掉列选择

    Flexigrid中会出现列选择的小黑箭头,有时挺讨厌,想去掉.发现没有控制的地方,于是自己加. 在flexigrid.js中增加 在定义中增加 p = $.extend({ //apply defa ...

  7. mysql基础:列类型--整型

    mysql列类型--字符串  http://blog.csdn.net/jk110333/article/details/9342301 mysql列类型--时间和日期    http://blog. ...

  8. Codeforces Round #197 (Div. 2) D. Xenia and Bit Operations

    D. Xenia and Bit Operations time limit per test 2 seconds memory limit per test 256 megabytes input ...

  9. js获取并设置&lt;p&gt;&lt;/p&gt;的显示的值。

    原文链接:http://www.nowamagic.net/librarys/posts/jquery/23 html()方法 此方法类似于JavaScript中的innerHTML属性,能够用来读取 ...

  10. Hibernate @Embeddable注解

    在hibernate中实现自己定义类型,仅仅要实现UserType接口就可以或者以Component的形式提供.JPA的@Embedded有点类似,通过此凝视能够在你的Entity中使用一般的java ...