SPOJ 1825 FTOUR2 - Free tour II (树上点分治)

原创 2015年07月10日 15:35:24

题目地址:SPOJ 1825
树分治的题果然除了模板题就是金牌题啊。。。这题是一道论文题,想了好长时间。。。。终于过了,,,,注意一个坑点,如果权值全部为负的话,是可以不选任意一条边的,这样权值为0。。。也就是说初始值要设为0。。。
具体看漆子超的论文《分治算法在树的路径问题中的应用》。。
代码如下:

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
#include <time.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
//#pragma comment(linker, "/STACK:1024000000")
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-9;
const int MAXN=200000+10;
int n, k, head[MAXN], cnt, root, min1, ans;
int F[MAXN];//F[i]表示在该子树之前所遍历的所有子树的路径上不超过i个黑点的路径的最长长度
int G[MAXN];//表示严格有i个黑点的最长路径
int siz[MAXN], color[MAXN], vis[MAXN], num[MAXN];
struct node
{
        int u, v, w, next;
}edge[MAXN<<1];
struct N
{
        int v, num, w;
}T[MAXN];
bool cmp(N x, N y)
{
        return x.num<y.num;
}
void add(int u, int v, int w)
{
        edge[cnt].v=v;
        edge[cnt].w=w;
        edge[cnt].next=head[u];
        head[u]=cnt++;
}
void init()
{
        memset(head,-1,sizeof(head));
        memset(color,0,sizeof(color));
        memset(vis,0,sizeof(vis));
        cnt=0;
}
void getroot(int u, int fa, int s)
{
        int i, max1=-1;
        for(i=head[u];i!=-1;i=edge[i].next){
                int v=edge[i].v;
                if(v==fa||vis[v]) continue ;
                getroot(v,u,s);
                max1=max(max1,siz[v]);
        }
        max1=max(max1,s-siz[u]);
        if(min1>max1){
                min1=max1;
                root=u;
        }
}
void getsize(int u, int fa)
{
        siz[u]=1;
        for(int i=head[u];i!=-1;i=edge[i].next){
                int v=edge[i].v;
                if(v==fa||vis[v]) continue ;
                getsize(v,u);
                siz[u]+=siz[v];
        }
}
void getnum(int u, int fa)
{
        num[u]=color[u];
        for(int i=head[u];i!=-1;i=edge[i].next){
                int v=edge[i].v;
                if(v==fa||vis[v]) continue ;
                getnum(v,u);
                num[u]=max(num[u],num[v]+color[u]);
        }
}
void getG(int u, int fa, int dep, int val)
{
        G[dep]=max(G[dep],val);
        for(int i=head[u];i!=-1;i=edge[i].next){
                int v=edge[i].v;
                if(v==fa||vis[v]) continue ;
                getG(v,u,dep+color[v],val+edge[i].w);
        }
}
void work(int u)
{
        vis[u]=1;
        int i, j;
        for(i=head[u];i!=-1;i=edge[i].next){
                int v=edge[i].v;
                if(vis[v]) continue ;
                getsize(v,-1);
                min1=INF;
                getroot(v,-1,siz[v]);
                work(root);
        }
        int tot=0;
        for(i=head[u];i!=-1;i=edge[i].next){
                int v=edge[i].v;
                if(vis[v]) continue ;
                getnum(v,-1);
                T[tot].v=v;
                T[tot].num=num[v];
                T[tot].w=edge[i].w;
                tot++;
        }
        sort(T,T+tot,cmp);
        int lim=k-color[u];
        for(i=0;i<=T[tot-1].num;i++) F[i]=-INF;
        for(i=0;i<tot;i++){
                for(j=0;j<=T[i].num;j++) G[j]=-INF;
                getG(T[i].v,u,color[T[i].v],T[i].w);
                if(i){
                        for(j=0;j<=T[i].num&&j<=lim;j++){
                                int tmp=min(lim-j,T[i-1].num);
                                if(F[tmp]==-INF) continue ;
                                ans=max(ans,F[tmp]+G[j]);
                        }
                }
                for(j=0;j<=T[i].num&&j<=lim;j++){
                        F[j]=max(F[j],G[j]);
                        if(j) F[j]=max(F[j],F[j-1]);
                        ans=max(ans,F[j]);
                }
        }
        vis[u]=0;
}
int main()
{
        int m, i, u, v, w, x;
        while(scanf("%d%d%d",&n,&k,&m)!=EOF){
                init();
                for(i=0;i<m;i++){
                        scanf("%d",&x);
                        color[x]=1;
                }
                for(i=1;i<n;i++){
                        scanf("%d%d%d",&u,&v,&w);
                        add(u,v,w);
                        add(v,u,w);
                }
                ans=0;
                getsize(1,-1);
                min1=INF;
                getroot(1,-1,n);
                work(root);
                printf("%d\n",ans);
        }
        return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

SPOJ FTOUR2 - Free tour II [点分治+启发式合并]【分治】

题目链接:http://www.spoj.com/problems/FTOUR2/ ————————————————————————————————————— FTOUR2 - Free tour...

Spoj FTOUR Free Tour II

Spoj FTOUR Free TourII 树的点分治

SPOJ FTOUR2 Free tour II

Description After the success of 2nd anniversary (take a look at problem FTOUR for more details...

SPOJFTOUR2-Free tour II

FTOUR2 - Free tour II no tags  After the success of 2nd anniversary (take a look at problem FTO...

spoj FTOUR2(树的分治)

题目链接 QZC树的分治论文第二道例题, 自己想了想觉得没什么思路, 后来看了论文中的解法看到中间时想到了一个线段树优化dp的解法, 论文中也提到的令dp[L] 标示从当前根结点向下经过L个crow...
  • tjdrn
  • tjdrn
  • 2013年08月17日 11:18
  • 888

【SPOJ】1825 Free tour II 点分治

传送门:【SPOJ】1825 Free tour II

树分治点分治(spoj1825 Free tour II)

看了别人的题解。。 题意: 有N个顶点的树,节点间有权值, 节点分为黑点和白点。 找一条最长路径使得 路径上黑点数量不超过K个 F[ i , j ] 表示它的第 I 个子树中经过的不...

【spoj 1825】Free tour II

【Problem】 After the success of 2nd anniversary (take a look at problem FTOUR for more details), thi...

PAT (Advanced Level) Practise 1100 Mars Numbers (20)

1100. Mars Numbers (20) 时间限制 400 ms 内存限制 65536 kB 代码长度限制 16000 B ...

HDU6201-transaction transaction transaction

transaction transaction transaction                                                                ...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:SPOJ 1825 FTOUR2 - Free tour II (树上点分治)
举报原因:
原因补充:

(最多只允许输入30个字)