正题
题目链接:https://gmoj.net/senior/#main/show/5248
题目大意
nnn个点的一棵树,有mmm张票(v,k,w)(v,k,w)(v,k,w)表示可以在点vvv买,可以走kkk步以内,价格为www。
一个时间只能有一个票,每次询问一个点到根节点需要的最少票钱。
解题思路
设fif_{i}fi表示iii到根节点的价钱,然后vectorvectorvector存票,倍增转移即可。
时间复杂度O(nlogn)O(n\log n)O(nlogn)
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define mp(x,y) make_pair(x,y)
using namespace std;
const int N=1e5+10,T=20;
struct node{int to,next;
}a[N*2];
int n,m,tot,ls[N],f[N][T],g[N][T];
vector<pair<int,int> > q[N];
void addl(int x,int y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;return;
}
void dfs(int x,int fa){for(int i=0;i<q[x].size();i++){int now=fa,k=q[x][i].first,w=q[x][i].second;for(int i=T-1;i>=0;i--)if((k>>i)&1)g[x][0]=min(g[x][0],g[now][i]+w),now=f[now][i];}for(int i=1;i<T;i++)g[x][i]=min(g[x][i-1],g[f[x][i-1]][i-1]),f[x][i]=f[f[x][i-1]][i-1];for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(y==fa)continue;f[y][0]=x;dfs(y,x);}return;
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<n;i++){int x,y;scanf("%d%d",&x,&y);addl(x,y);addl(y,x);}for(int i=1;i<=m;i++){int x,y,w;scanf("%d%d%d",&x,&y,&w);q[x].push_back(mp(y,w));}memset(g,0x3f,sizeof(g));g[1][0]=0;dfs(1,1);scanf("%d",&m);while(m--){int x;scanf("%d",&x);printf("%d\n",g[x][0]);}
}