【dp】【倍增】花花的聚会

4.1 题意描述
花花住在H 国。H 国有n 个城市,其中1 号城市为其首都。城市间有n �� 1 条单向道
路。从任意一个城市出发,都可以沿着这些单向道路一路走到首都。事实上,从任何一个城市
走到首都的路径是唯一的。
过路并不是免费的。想要通过某一条道路,你必须使用一次过路券。H 国一共有m 种过
路券,每张过路券以三个整数表示:v k w:你可以在城市v 以价格w 买到一张过路券。这
张券可以使用k 次。这意味着,拿着这张券通过了k 条道路之后,这张券就不能再使用了。
请注意你同一时间最多只能拥有最多一张过路券。但你可以随时撕掉手中已有的过路券,
并且在所在的城市再买一张。
花花家在首都。他有q 位朋友,他希望把这些朋友们都邀请到他家做客。所以他想要知道
每位朋友要花多少路费。他的朋友们都很聪明,永远都会选择一条花费最少的方式到达首都。
花花需要准备晚餐去了,所以他没有时间亲自计算出朋友们将要花费的路费。你可以帮帮
他么?
4.2 输入格式
输入的第一行包含两个空格隔开的整数n 和m,表示H 国的城市数量和过路券的种数。
之后的n �� 1 行各自包含两个数ai 和bi,代表城市ai 到城市bi 间有一条单向道路。
之后的m 行每行包括三个整数vi; ki 和wi,表示一种过路券。
下一行包含一个整数q,表示花花朋友的数量。
之后的q 行各自包含一个整数,表示花花朋友的所在城市。
4.4 样例输入
7 7
3 1
2 1
7 6
6 3
5 3
4 3
7 2 3
7 1 1
2 3 5
3 6 2
4 2 4
5 3 10
6 1 20
3
5
6
7
4.5 样例输出
10
22
5
4.6 样例解释
对于第一位朋友,他在5 号城市只能购买一种过路券,花费10 元并且可以使用3 次。这
足够他走到首都,因此总花费是10 元。
对于第二位朋友,他在6 号城市只能购买20 元的过路券,并且只能使用一次。之后,他
可以在3 号城市购买2 元,可以使用3 次的过路券走到首都。总花费是22 元。
对于第三位朋友,他在7 号城市可以购买两种过路券。他可以花3 元买一张可以使用2
次的券,然后在3 号城市再买一张2 元,可以使用3 次的券,走到首都。总花费是5 元,而且
其他的购买方式不会比这种更省钱。
4.7 数据规模与约定
• 对于40% 的数据:n; m; q 10;wi 10;
• 另有20% 的数据:n; m; q 500;wi 100;
• 另有20% 的数据:n; m; q 5000;wi 1000;
• 对于100% 的数据:n; m; q 105;wi 10000; 1 vi; ki n。

很明显是一个DP题。首相容易想到转移方程dp[v]=min{dp[u]+w}
其中u为v祖先,w为可以从v买的票的价值,初值dp[1]=0,以根节点DFS往下DP即可,O(mn),会超时。那我们就可以倍增在树上维护minn[dp[u]],即可AC

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#ifdef WIN32 
#define AUTO "%I64d"
#else 
#define  AUTO "%lld"
#endif 
#define long long LL
using namespace std;
const int maxn=5e5+5,INF=0x7f7f7f7f;
int n,m,q,to[maxn*2],next[maxn*2],head[maxn],To[maxn],cost[maxn],Next[maxn],Head[maxn],dp[maxn],cnt,edge;
int fa[maxn][20],dis[maxn][20];
void edge_add(int u,int v)
{
    to[++edge]=v;
    next[edge]=head[u];
    head[u]=edge;
}
void ticket_add(int u,int k,int w)
{
    To[++cnt]=k;
    cost[cnt]=w;
    Next[cnt]=Head[u];
    Head[u]=cnt;
}
void dfs(int u,int FA)
{
    fa[u][0]=FA; dis[u][0]=dp[FA];
    for (int i=1;i<=16;i++)
    {
        fa[u][i]=fa[fa[u][i-1]][i-1];
        dis[u][i]=min(dis[u][i-1],dis[fa[u][i-1]][i-1]);
    }
    for (int T=Head[u];T;T=Next[T])
    {
        int k=To[T],w=cost[T],pos=u,minn=INF;
        for (int i=0;i<=16;i++)
        {
            if (k&(1<<i))
            {
                minn=min(minn,dis[pos][i]);
                pos=fa[pos][i];
            }   
        }
        dp[u]=min(dp[u],w+minn);
    }
    for (int E=head[u];E;E=next[E])
    {
        int v=to[E];
        if (v==FA) continue;
        dfs(v,u);
    }
}
int main()
{
    freopen("party.in","r",stdin);
    freopen("party.out","w",stdout);
    memset(dp,INF,sizeof(dp));
    memset(dis,INF,sizeof(dis));
    dp[1]=0;
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        edge_add(u,v); edge_add(v,u);
    }
    for (int i=1;i<=m;i++)
    {
        int u,k,w;
        scanf("%d%d%d",&u,&k,&w);
        ticket_add(u,k,w);
    }
    dfs(1,0);
    scanf("%d", &q);
    for (int i = 1; i <= q; i++) 
    {
        int u;
        scanf("%d",&u);
        printf("%d\n", dp[u]);
    }
    return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值