初学分层图,写写模板题。
个人感觉分层图大部分会应用在最短路上,主要体现为题目会要求将图中的一些边的边权进行改变。比如选择 k k k条边不计算其贡献等等。这类题目的解题方法大部分都是在原先的 d j dj dj最短路算法上再增加几维状态,就像把这个图分为了好几层,每一层都由当前层或前一层的状态转移而来。
(以上纯属菜鸡理解,大佬勿喷)
题解:
因为要选择几条边不计贡献,那么我们就再增加一维状态, d i s [ i ] [ j ] dis[i][j] dis[i][j]表示达到点 i i i 时,已经选择了 j j j 条边不计入贡献的最短路。那么在跑最短路的时候就很好转移了,当前边要么不计入贡献, d i s [ v ] [ j ] = d i s [ u ] [ j ] + w dis[v][j]=dis[u][j]+w dis[v][j]=dis[u][j]+w ,要么计入贡献 d i s [ v ] [ j + 1 ] = d i s [ u ] [ j ] dis[v][j+1]=dis[u][j] dis[v][j+1]=dis[u][j]
代码:
#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int MAXN=2e5+5;
const int inf=0x3f3f3f3f;
int head[MAXN];
int vis[MAXN][15];
ll dis[MAXN][15];
int cnt;
struct edge
{
int to;
int w;
int next;
}e[MAXN*2];
struct node
{
int pos;
ll val;
int num;
friend bool operator<(node a,node b)
{
return a.val>b.val;
}
};
void add(int u,int v,int w)
{
e[cnt].to=v;
e[cnt].w=w;
e[cnt].next=head[u];
head[u]=cnt++;
}
void solve(int s,int k)
{
memset(vis,0,sizeof vis);
memset(dis,0x3f3f3f3f3f3f3f3f,sizeof dis);
priority_queue<node> q;
q.push(node{s,0,0});
dis[s][0]=0;
while(!q.empty())
{
node now=q.top();
q.pop();
if(vis[now.pos][now.num]) continue;
vis[now.pos][now.num]=1;
for(int i=head[now.pos];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(dis[v][now.num]>now.val+e[i].w)
{
dis[v][now.num]=now.val+e[i].w;
q.push({v,dis[v][now.num],now.num});
}
if(now.num<k&&dis[v][now.num+1]>now.val)
{
dis[v][now.num+1]=now.val;
q.push({v,dis[v][now.num+1],now.num+1});
}
}
}
}
int main()
{
int n,m,k;
cin>>n>>m>>k;
int s,t;
cin>>s>>t;
memset(head,-1,sizeof head);
for(int i=1;i<=m;i++)
{
int u,v,w;
cin>>u>>v>>w;
add(u,v,w);
add(v,u,w);
}
solve(s,k);
ll ans=1e18;
for(int i=0;i<=k;i++)
{
ans=min(ans,dis[t][i]);
}
cout<<ans<<endl;
}