PAT 甲级 1018 Public Bike Management

1018 Public Bike Management

解题思路

对体面进行分析，可以很直观地感受到可以使用dijkstra算法和dfs算法结合求解。
先使用dijkstra算法确定最短路径，然后使用dfs算法进行模拟调整的过程，确定出minNeed和minBack，使结果满足最优子结构。

代码

#include<bits/stdc++.h>
#include<iostream>
using namespace std; 
const int maxn=510;
const int INF=100000000;
int G[maxn][maxn];
int d[maxn];
int vis[maxn];
int needs[maxn];
int C,N,S,M;
int temp[maxn]={0},path[maxn]={0};
int mindis,mintake=INF,minremain=INF;
int pre[maxn][maxn];
int prenum[maxn];
 
void Dijkstra(int end){
int i,j;
int u,min,v;
d[0]=0;
for(i=0;i<=N;i++){
u=-1,min=INF;
for(j=0;j<=N;j++){
if(vis[j]==0&&d[j]<min){
u=j;
min=d[u];
}
}
if(u==-1){
break;
}
vis[u]=1;
for(v=0;v<=N;v++){
if(vis[v]==0&&G[u][v]!=INF){
if(d[u]+G[u][v]<d[v]){
d[v]=d[u]+G[u][v];
prenum[v]=0;
pre[v][prenum[v]++]=u;
}else if(d[u]+G[u][v]==d[v]){
pre[v][prenum[v]++]=u;
}
}
}
if(u==end){
mindis=d[end];
return;
}
}
}
 
void DFS(int u,int depth){
int i,j=0;
int take,remain,v;
temp[depth]=u;
if(u==0){
take=0,remain=0;
for(i=depth-1;i>=0;i--){
v=temp[i];
if(remain>=needs[v]){
remain=remain-needs[v];
}else{
take+=needs[v]-remain;
remain=0;
}
} 
if(take<mintake){
mintake=take;
minremain=remain;
for(i=depth;i>=0;i--){
path[j++]=temp[i];
}
}else if(take==mintake&&remain<minremain){
mintake=take;
minremain=remain;
for(i=depth;i>=0;i--){
path[j++]=temp[i];
}
}
return;
}
for(i=0;i<prenum[u];i++){
v=pre[u][i];
DFS(v,depth+1);
}
}
 
int main(){
int t;
int i,j;
int a,b,v;
cin>>C>>N>>S>>M; 
C=C/2;
for(i=1;i<=N;i++){
cin>>needs[i]; 
needs[i]=C-needs[i];
}
for(i=0;i<=N;i++){
d[i]=INF;
vis[i]=0;
for(j=0;j<=N;j++){
G[i][j]=INF;
}
G[i][i]=0;
}
for(i=0;i<M;i++){
cin>>a>>b>>v;
G[a][b]=G[b][a]=v;
}
Dijkstra(S);
DFS(S,0);
cout<<mintake<<" 0";
for(t=1;;t++){
cout<<"->"<<path[t];
if(path[t]==S){
break;
}
}
cout<<" "<<minremain<<endl;
return 0;
}