杭电多校#5

2022杭电多校5

1003 Slipper

题意： 给出一棵n点构成的树，树上每条边都有边权wi，表示移动所需的能量，对于深度差为k的点还可以通过花费p点能量传送过去，给出起点s和终点t，问从起点到达终点所需最少能量。

思路： 对于深度差为k的点对建边，跑最短路但是TLE，对于这两层建立一个虚点，各层上的点都连到虚点上，这样就会少添加很多边，对于向上传送建立一个虚点，对于向下传送还需要建立一个虚点，建好图dijkstra；

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int N=1e6+10;
typedef pair<long long,int> PII;
int depth[N];
long long d[3*N];
int maxd,be,en,n;
vector<int>p[N];
int h[N*3],e[N*10],ne[N*10],idx;
long long w[N*10];
bool vis[3*N];
void add(int x,int y,int z)
{
	e[idx]=y;
	w[idx]=z;
	ne[idx]=h[x];
	h[x]=idx++;
}
void dfs(int x,int fa)
{
	p[depth[x]].push_back(x);
	for(int i=h[x];i!=-1;i=ne[i])
	{
		int j=e[i];
		if(j==fa) continue;
		depth[j]=depth[x]+1;
		maxd=max(maxd,depth[j]);
		dfs(j,x);
	}
}
void dijkstra()
{
    for(int i=1;i<=3*n;i++)
    {
    	vis[i]=false;
    	d[i]=0x3f3f3f3f3f3f3f3f;
	}
    d[be]=0;
    priority_queue<PII,vector<PII>,greater<PII> >q;
    q.push({0,be});
    while(!q.empty())
    {
        int begin=q.top().second;
        q.pop();
        if(begin==en) return ;
        if(vis[begin]) continue;
        vis[begin]=true;
        for(int i=h[begin];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(d[j]>d[begin]+w[i])
            {
                d[j]=d[begin]+w[i];
                q.push({d[j],j});
            }
        }
    }
}
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		scanf("%d",&n);
		idx=maxd=0;
		for(int i=1;i<=3*n;i++) h[i]=-1;
		for(int i=1;i<n;i++)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			add(u,v,w);add(v,u,w);
		}
		int k,val;
		scanf("%d%d",&k,&val);
		scanf("%d%d",&be,&en);
		depth[1]=1;
		dfs(1,-1);
		for(int i=1;i<=maxd;i++)
		for(int j=0;j<p[i].size();j++)
		{
			add(p[i][j],n+i,0);
			add(2*n+i,p[i][j],0);
		}
		for(int i=1;i+k<=maxd;i++)
		{
			add(n+i,2*n+k+i,val);
			add(n+k+i,2*n+i,val);
		}
		dijkstra();
		printf("%lld\n",d[en]);
		for(int i=1;i<=maxd;i++) p[i].clear();
	}
	return 0;
}

1007 Count Set

思路：暂无