题目链接https://pintia.cn/problem-sets/994805342720868352/problems/994805489282433024
这道题真的拖了好久好久啊,今天终于写完了。感觉dij求最短路越来越熟练了,但是dfs还是有点勉强。看了柳神的代码,终于完全理解了。然后凭着自己的理解敲了一遍,一发过!开心!希望每天进步都能那么一点点!冲冲冲!
若有错误,还望大神指正!
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e3+10;
const int INF=0x3f3f3f3f;
int k,n,t,m;//每个点的最大容量、站点数量、问题站点、路径数量
int c[maxn];//每个站点的自行车数量
int vis[maxn];//标记站点是否已被访问
int dis[maxn];//从起点(PMBC)到某点的最短距离
int e[maxn][maxn];//存图
int minneed=INF,minback=INF;//需从PBMC带走的最小自行车数量、最终带回PBMC的最小自行车数量
vector<int>pre[maxn];//记录最短路中,某点的前驱站点.如pre[3]={1,2};pre[1]=0;pre[2]=0;表示从起点到站点3有两条最短路,分别为0->1->3和0->2->3;
vector<int>path,temppath;
void dfs(int v){
temppath.push_back(v);//获取最短路
if(v==0){//当搜索到v==0即起点(PMBC)时,temppath中已“得到”了一条最短路
int need=0,back=0;
for(int i=temppath.size()-1;i>=0;i--){//遍历最短路上的站点
int id=temppath[i];
if(c[id]>0){//若站点自行车数量超限
back+=c[id];//需带回多余部分
}
else{//站点自行车数量不足
if(back>(0-c[id])){//准备带回的自行车足以弥补
back+=c[id];//从要带回的自行车中扣除
}
else{//准备带回的自行车不足以弥补
need+=(0-c[id]-back);//扣除要带回的自行车后,还需从PBMC再拿
back=0;//此时没有可以带回的自行车了
}
}
}
if(need<minneed){//找最小值
minneed=need;
minback=back;
path=temppath;
}
else if(need==minneed&&back<minback){//若need相同,则选back较小的
minback=back;
path=temppath;
}
}
for(int i=0;i<pre[v].size();i++){
dfs(pre[v][i]);
}
temppath.pop_back(); //清除最后一个元素,寻找下一条最短路
}
//dij单源最短路模板
void dij(){
memset(vis,0,sizeof(vis));
memset(dis,INF,sizeof(dis));
for(int i=0;i<=n;i++){
dis[i]=e[0][i];
}
for(int i=1;i<=n;i++){
int mmin=INF,u=-1;
for(int j=0;j<=n;j++){
if(vis[j]==0&&dis[j]<mmin){
mmin=dis[j];
u=j;
}
}
if(u==-1) break;
vis[u]=1;
for(int j=0;j<=n;j++){
if(vis[j]==0&&dis[u]+e[u][j]<dis[j]){
//更新最短路,以及站点的前驱结点
dis[j]=dis[u]+e[u][j];
pre[j].clear(); //经u到j的是最短路,之前记录的东西就全部清空(因为不是最短的)
pre[j].push_back(u);
}
else if(vis[j]==0&&dis[u]+e[u][j]==dis[j]){
//多条最短路的情况
pre[j].push_back(u);
}
}
}
}
int main()
{
cin>>k>>n>>t>>m;
for(int i=1;i<=n;i++){
cin>>c[i];
c[i]=c[i]-k/2; //若c[i]>0,说明此站点的自行车数量大于最大容量的一半,需要拿走c[i]辆自行车,否则需要补充0-c[i]辆自行车
}
for(int i=0;i<=n;i++){
for(int j=0;j<=n;j++){//初始化
if(i==j) e[i][j]=0;
else e[i][j]=INF;
}
}
while(m--){
int u,v,w;
cin>>u>>v>>w;
if(w<e[u][v]){
e[u][v]=w;
e[v][u]=w;
}
}
dij();//寻找最短路
dfs(t);//从问题站点开始搜索
cout<<minneed<<" 0";
for(int i=path.size()-2;i>=0;i--){
cout<<"->"<<path[i];
}
cout<<" "<<minback<<endl;
}