YbtOJ 小明聚会(dp+倍增优化)

5min就能想到,设置 f i f_{i} fi 表示从i到1的最少花费
f i = m i n ( f j + c o s t ) ( d e p i − d e p j ≤ k & i = s t ) f_{i}=min(f_{j}+cost) (dep_i-dep_j\le k\& i=st) fi=min(fj+cost)(depidepjk&i=st)
那可以搞一个栈去维护 d e p i − d e p j ≤ k dep_i-dep_j\le k depidepjk
诶,不对啊,这是倍增题单啊, O ( n 2 ) O(n^2) O(n2) 能过??

请添加图片描述

哦,tmd真过了
再写一下倍增,怎么回事呢,我 O ( n 2 ) O(n^2) O(n2) O ( n log ⁡ n ) O(n\log n) O(nlogn) 都快???内存还小????

请添加图片描述

YbtOJ你演我????
//O(n^2)
#include<bits/stdc++.h>
using namespace std;
const int maxn=3e5+10;
struct TICKET{int st,cost,k;}t[maxn];
struct ED{int nxt,to;}e[maxn];
int edgenum,head[maxn],n,m,top,sta[maxn],f[maxn];
bool vis[maxn];
vector<int>q[maxn];
inline void add(int x,int y)
{
    e[++edgenum].nxt=head[x];
    head[x]=edgenum;
    e[edgenum].to=y;
}
inline void dfs(int x)
{
    vis[x]=1,sta[++top]=x;
    for(int i=head[x];i;i=e[i].nxt)
    {
        int y=e[i].to;
        if(vis[y]) continue;
        if(!q[y].empty())
            for(auto tmp:q[y])
                for(int j=top;j>=max(top-t[tmp].k+1,1);j--)
                        f[y]=min(f[y],f[sta[j]]+t[tmp].cost);
        dfs(y);
    }top--;
}
signed main()
{
    memset(f,0x3f,sizeof(f));
    cin>>n>>m,f[1]=0;
    for(int i=1,x,y;i<n;i++)
        cin>>x>>y,add(x,y),add(y,x);
    for(int i=1;i<=m;i++)
        cin>>t[i].st>>t[i].k>>t[i].cost,q[t[i].st].push_back(i);
    dfs(1);
    int qi;cin>>qi;
    for(int i=1,x;i<=qi;i++) cin>>x,cout<<f[x]<<endl;
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值