BJ 集训8 测试 B 新访问计划

http://www.elijahqi.win/archives/2757
题意：参考apio2010巡逻 只不过这次将可以修建的道路增加到2e4

那么考虑增加枚举每次跳跃的花费 那么可以知道随着跳跃的花费增加我的跳跃次数是递减的

所以那么考虑显然如果我跳跃的答案更小的话 当我跳跃次数<=k的时候我一定尽量跳跃更多的即可 那么设dp数组dp[x][0/1]分别表示当前我处在x节点 我没有走回/走回当前节点 我的最优答案是多少 在满足这个最优答案的时候我需要跳跃几次才可以满足条件 注意当结尾的时候我可能存在一些问题 比如我跳跃的时候代价可能正好等于我走边的边权的时候 那么我可能用跳跃来代替了我这个走边权的机会那么因为我跳跃固定在c 所以我就会多增加一些费用 所以应该恰好将k条边多增加的代价卡下来

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 22000
#define pa pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define inf 0x3f3f3f3f
using namespace std;
inline int read(){
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch<='9'&&ch>='0') x=x*10+ch-'0',ch=getchar();
    return x;
}
struct node{
    int y,next,z;
}data[N<<1];
pa dp[N][2];
int h[N],n,k,c,cost,num;
inline void dfs(int x,int fa){
    dp[x][0].fi=dp[x][1].fi=dp[x][0].se=dp[x][1].se=0;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y,z=data[i].z;if (y==fa) continue;pa tmp0,tmp1;dfs(y,x);
        tmp0=min(mp(dp[x][1].fi+dp[y][0].fi,dp[x][1].se+dp[y][0].se),mp(dp[x][0].fi+dp[y][1].fi+z,dp[x][0].se+dp[y][1].se));
        tmp1=min(mp(dp[x][1].fi+dp[y][1].fi+z,dp[x][1].se+dp[y][1].se),mp(dp[x][0].fi+dp[y][0].fi+cost,dp[x][0].se+dp[y][0].se+1));
        dp[x][0]=tmp0;dp[x][1]=tmp1;
    }
}
inline bool check(int x){
    cost=x;dfs(1,1);return dp[1][1].se<=k;
}
int main(){
    freopen("t2.in","r",stdin);
    while(~scanf("%d",&n)){
        k=read();c=read();
        memset(h,0,sizeof(int)*(n+10));num=0;int sum=0;
        for (int i=1;i<n;++i) {
            int x=read()+1,y=read()+1,z=read();sum+=z;
            data[++num].y=y;data[num].next=h[x];data[num].z=z;h[x]=num;
            data[++num].y=x;data[num].next=h[y];data[num].z=z;h[y]=num;
        }cost=c;dfs(1,1);if (dp[1][1].se<=k) {printf("%d\n",dp[1][1].fi+sum);continue;}
        int l=c+1,r=1e9,mid;
        while(l<=r){
            mid=l+r>>1;
            if (check(mid)) r=mid-1;else l=mid+1;
        }check(l);printf("%d\n",sum+dp[1][1].fi-(cost-c)*k);
    }
    return 0;
}