bzoj2599/洛谷4149 [IOI2011] Race(树分治)

2人阅读 评论(0) 收藏 举报
分类:

题意:

洛谷的地址
给一棵树,每条边有权。求一条简单路径,权值和等于 K ,且边的数量最小。

思路:

这题bzoj下线了,就去洛谷上A的
其实是一个比较简单的点分治
但是这题和之前的点分治不太一样(这题),这里的要求和之前不一样,我们不能算完重心的情况再去算子树的情况了
我们换一个思路,我们先遍历一颗以重心的儿子为根的子树,然后和一个之前的num(num[i]表示,到重心距离为i时最少经过几个边)数组来更新答案,更新完后,再把这棵子树的东西放进num数组,这样就能很好地排除非法情况(就是两点的简单路径不经过重心)
之后就和正常点分治一样了

错误及反思:

其实上面的都很好想,也不难写。然后就开始十分诡异的RE,TLE了,我在统计每颗子树的时候,把fa写成-1是tle,写成now却是re,然后就找了一下午bug,重写了一次,把vector改成手动模拟,也还是过不了
后来看了看别人的写法,把num数组的清空方式改了一下(之前是每次都memset,后来改成再重新跑一次,把经过的地方都修改回去),就AC了,甚至fa写错了都能AC。。。
可是我们计算一下复杂度,每次都memset的复杂度是O(logn(n+memset(1e6))),这样是tle或re,改了以后是O(logn(n+n)),可是memset很快的啊,因为这个tle实在是搞不懂了。。。re就更神秘了

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<vector>
#define fi first
#define se second
using namespace std;
const int N = 200100;
struct EDGE{
    int to,next,val;
}e[N*2];
int first[N],n,tot=0,si[N],maxn[N],num[1000010],k,ans=1e9,v[N][2],en;
bool did[N];

void addedge(int x,int y,int z){
    e[tot].to=y;
    e[tot].val=z;
    e[tot].next=first[x];
    first[x]=tot++;
    e[tot].to=x;
    e[tot].val=z;
    e[tot].next=first[y];
    first[y]=tot++;
}
void dfs_size(int now,int fa){
    si[now]=1;
    maxn[now]=0;
    for(int i=first[now];i!=-1;i=e[i].next)
        if(e[i].to!=fa&&!did[e[i].to]){
            dfs_size(e[i].to,now);
            si[now]+=si[e[i].to];
            maxn[now]=max(maxn[now],si[e[i].to]);
        }
}
void dfs_root(int now,int fa,int& root,int& nu,int t){
    int MA=max(maxn[now],si[t]-si[now]);
    if(MA<nu){
        nu=MA;
        root=now;
    }
    for(int i=first[now];i!=-1;i=e[i].next)
        if(e[i].to!=fa&&!did[e[i].to])
            dfs_root(e[i].to,now,root,nu,t);
}
void dfs2(int now,int fa,int tlen,int dep){
    if(tlen>k) return ;
    for(int i=first[now];i!=-1;i=e[i].next)
        if(e[i].to!=fa&&!did[e[i].to])
            dfs2(e[i].to,now,tlen+e[i].val,dep+1);
    v[en][0]=tlen;
    v[en++][1]=dep;
}

void solve(int now){
    int root,nu=1e9;
    dfs_size(now,-1);
    dfs_root(now,-1,root,nu,now);
    num[0]=0;
    did[root]=true;
    for(int i=first[root];i!=-1;i=e[i].next){
        if(!did[e[i].to]){
            en=0;
            dfs2(e[i].to,root,e[i].val,1);//很神秘的地方,把fa写成-1是tle,写成now却是re,换了num的写法后却都能ac。。。
            for(int j=0;j<en;j++)
                ans=min(ans,v[j][1]+num[k-v[j][0]]);
            for(int j=0;j<en;j++)
                num[v[j][0]]=min(num[v[j][0]],v[j][1]);
        }
    }
    for(int i=first[root];i!=-1;i=e[i].next){
        if(!did[e[i].to]){
            en=0;
            dfs2(e[i].to,root,e[i].val,1);

            for(int j=0;j<en;j++)
                num[v[j][0]]=1e9;
        }
    }
    for(int i=first[root];i!=-1;i=e[i].next)
        if(!did[e[i].to])
            solve(e[i].to);
}
int main(){
    memset(first,-1,sizeof(first));
    scanf("%d%d",&n,&k);
    for(int i=1;i<=k;i++) num[i]=1e9;//这里只写一次,而不是每次都memset
    for(int i=0,u,v,w;i<n-1;i++){
        scanf("%d%d%d",&u,&v,&w);
        addedge(u+1,v+1,w);
    }
    solve(1);
    if(ans==1e9) printf("-1\n");
    else printf("%d\n",ans);
}
查看评论

[BZOJ2599][IOI2011]Race 树分治

题意求树上权值和为K的最小简单路径树分治每次分治找到的Root,T[x]表示离Root距离为x的最小边数量,DFS时更新答案。表示T数组开小竟然是TLE很不科学……#include #include...
  • Coldef
  • Coldef
  • 2017-01-25 22:32:25
  • 223

bzoj2599【IOI2011】Race

树的点分治
  • AaronGZK
  • AaronGZK
  • 2016-05-13 00:26:14
  • 1758

【bzoj2559】【IOI2011】【Race】【点分治】

Description 给一棵树,每条边有权.求一条路径,权值和等于K,且边的数量最小. Input 第一行 两个整数 n, k 第二..n行 每行三个整数 表示一条无向边的两...
  • sunshinezff
  • sunshinezff
  • 2015-09-27 13:28:14
  • 1388

【bzoj2599】[IOI2011]Race 点分治

点分治 1、求树的重心 2、计算以当前重心为根的子树的答案 3、去掉以当前重心儿子为根的子树的答案 4、枚举每个儿子,分治 考虑计算过程如何实现 我们不妨记一个ans数组,ans[i]...
  • u012288458
  • u012288458
  • 2016-03-18 09:04:34
  • 1479

2599: [IOI2011]Race|树的点分治

数据范围:n≤200000,K≤1000000n\leq200000,K\leq1000000(题意描述中怎么没有。。 然后就是分治,找经过根的路径开数组存下距离为xx时的最小边数cnt[x]cnt...
  • ws_yzy
  • ws_yzy
  • 2016-03-13 15:41:27
  • 378

bzoj 2599: [IOI2011]Race

Description 给一棵树,每条边有权.求一条路径,权值和等于K,且边的数量最小. Input 第一行 两个整数 n, k 第二..n行 每行三个整数 表示一条无向边的两...
  • lqybzx
  • lqybzx
  • 2015-04-02 10:47:28
  • 839

【IOI2011】bzoj2599 Race

点分治
  • sdfzyhx
  • sdfzyhx
  • 2017-02-14 14:48:51
  • 222

【bzoj2599】[IOI2011]Race

Description给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N ...
  • w_yqts
  • w_yqts
  • 2017-05-10 21:03:58
  • 78

【bzoj2599】: [IOI2011]Race

sb的我一开始想统计每棵子树中经过根的路径有多少条应该比较容易 就把休息站选成根就行了。。 你会发现其实这样的话在每棵子树中到根不满足条件的点对完全没有统计 所以在每棵子树x中我们用f【i】表示以x...
  • Qingegege
  • Qingegege
  • 2016-02-24 14:41:20
  • 405

bzoj2599 [IOI2011]Race

Description给一棵树,每条边有权.求一条简单路径,权值和等于 KK ,且边的数量最小. N≤200000,K≤1000000N \leq 200000, K \leq 1000000Inpu...
  • aziint
  • aziint
  • 2017-12-01 20:31:25
  • 40
    个人资料
    持之以恒
    等级:
    访问量: 4927
    积分: 830
    排名: 6万+