杭州市有一个公共自行车服务中心,为来自世界各地的游客提供极大的便利。你可以在任何一个车站租一辆自行车,然后把它送回城里的任何一个车站。
公共自行车管理中心(PBMC)持续监控所有车站的实时通行能力。据说,如果一个车站正好是半满的,那么它就处于完美的状态。如果一个车站是满的或空的,人民广播公司将收集或发送自行车,以调整该车站的条件,以完善。而且,途中的所有车站也将进行调整。
当一个问题站点被报告时,PBMC总是选择到达该站点的最短路径。如果有多条最短路径,则选择需要从PBMC发送的最少自行车数的路径。有多条相同的发送车数路径时,选择带回PBMC最少车数的路。
先dijkstra找出最短路径,存储一个新图。然后dfs找出每条PBMC到sp的路径,计算带去和带回的车数。最后选一条最佳的路径输出。
很多时候都想去找捷径或者替代的做法,但是最后总是会发现按着它本来的思路才能ac
#include <bits/stdc++.h>
#define INF 9999999
using namespace std;
int m,n,cmax,sp;
const int maxn=510;
int g[maxn][maxn];
int num[maxn],d[maxn],take[maxn];
bool vis[maxn];
vector<int> gnew[maxn];
int takenum=INF,backnum=INF;
vector<int> path,res;
void dij(){
fill(vis,vis+n+1,false);
fill(d,d+n+1,INF);
d[0]=0;
take[0]=0;
for(int i=1;i<=n;i++){
int u=-1,MIN=INF;
for(int j=0;j<=n;j++){
if(!vis[j] && d[j]<MIN){
u=j;
MIN=d[j];
}
}
if(u==-1)return;
vis[u]=true;
for(int v=1;v<=n;v++){
if(!vis[v] && g[u][v]!=INF){
if(d[u]+g[u][v]<d[v]){
d[v]=d[u]+g[u][v];//更新距离
gnew[v].clear();
gnew[v].push_back(u);
}else if(d[u]+g[u][v]==d[v]){
gnew[v].push_back(u);
}
}
}
}
}
void dfs(int curr){
path.push_back(curr);
if(curr==0){
int taketo=0,takeback=0;
int len=path.size();
for(int i=len-2;i>=0;i--){
if(num[path[i]]<cmax){
takeback-=(cmax-num[path[i]]);//先从现有的车数中减
if(takeback<0){
taketo+=(-takeback);//保证车数不能小于0
takeback=0;
}
}else takeback+=(num[path[i]]-cmax);
}
if(taketo<takenum ||(taketo==takenum && takeback<backnum)){
takenum=taketo;
backnum=takeback;
res=path;
}
return;
}
for(int i=0;i<gnew[curr].size();i++){
dfs(gnew[curr][i]);
path.pop_back();
}
}
int main(){
scanf("%d%d%d%d",&cmax,&n,&sp,&m);
cmax=cmax/2;
for(int i=1;i<=n;i++){
scanf("%d",&num[i]);
}
fill(g[0],g[0]+maxn*maxn,INF);
for(int i=0;i<m;i++){
int a,b,dist;
scanf("%d%d%d",&a,&b,&dist);
g[a][b]=g[b][a]=dist;
}
dij();
dfs(sp);
printf("%d 0",takenum);
int cnt=0,len=res.size();
for(int i=len-2;i>=0;i--){
printf("->%d",res[i]);
}
printf(" %d",backnum);
}