PAT甲级-1018 Public Bike Management
题目链接
dijkstra的变式:除了最短路径外还要在这个基础上使得bring的车最少,send_back的车最少。
只用dijkstra第7个测试点会报错。因为bring 和send_back 不满足最优子结构。
题解
dijkstra+DFS
先筛出到目的地的最短路,可能存在多条路径。vector v 里头放的是每个节点可能的父节点。比如v[x].size()=3。表示x节点可能有三个父节点过来。再用dfs遍历每一条路径,更新bring 和send_back值。
- 如果最后一个测试点超时,需要把每一个v[i]去重。
- bring 的意思要弄清楚,比如 -5,5,0。(负数代表还要取车,正数代表要送回车),到第2个节点是bring =5,send_back=0。bring不能用后面的抵消前面的,send_back可以用前面的抵消后面的。
#include<iostream>
#include<vector>
#include<queue>
#include<string>
#include<algorithm>
#include<string.h>
using namespace std;
const int N=5e5;
int h[N],e[N],w[N],ne[N],top;
typedef pair<int,int>PII;
priority_queue<PII>q;
int curBike[N];
int capacity,n,d,m;
void add(int a,int b,int c){
e[++top]=b;
w[top]=c;
ne[top]=h[a];
h[a]=top;
}
int vis[N],dis[N];
vector<int>v[N];
void dijkstra(){
q.push({0,0});
while(!q.empty()){
int x=q.top().second;
int distance=-q.top().first;
q.pop();
if(vis[x])
continue;
vis[x]=1;
for(int j=h[x];j!=-1;j=ne[j]){
int y=e[j];
if(vis[y])
continue;
if(dis[y]>distance+w[j]){
dis[y]=distance+w[j];
v[y].clear();
v[y].push_back(x);
}
if(dis[y]==distance+w[j]){
v[y].push_back(x);
}
q.push({-dis[y],y});
}
}
}
void init(){
top=0;
for(int i=0;i<N;i++)
h[i]=-1;
curBike[0]=capacity/2;
for(int i=0;i<N;i++)
dis[i]=2e9;
}
vector<int>path,ans;
int bring=2e9,send_back=2e9;
void dfs(int x){//遍历到节点x
if(x==0){
int tbring=0,tsend=0;
for(int i=path.size()-1;i>=0;i--){
int y=path[i];
if(capacity/2-curBike[y]-tsend>0){
tbring+=capacity/2-curBike[y]-tsend;
tsend=0;
}else{
tsend=curBike[y]+tsend-capacity/2;
}
}
if(tbring<bring||(tbring==bring&&tsend<send_back)){
ans=path;
bring=tbring;
send_back=tsend;
}
return ;
}
sort(v[x].begin(),v[x].end());
int len=unique(v[x].begin(),v[x].end())-v[x].begin();
for(int i=0;i<len;i++){
int y=v[x][i];
path.push_back(y);
dfs(y);
path.pop_back();
}
}
int main(){
cin>>capacity>>n>>d>>m;
for(int i=1;i<=n;i++){
cin>>curBike[i];
}
init();
for(int i=0;i<m;i++){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
dijkstra();
path.push_back(d);
dfs(d);
cout<<bring<<" ";
for(int i=ans.size()-1;i>=0;i--){
if(i==ans.size()-1){
cout<<ans[i];
}else{
cout<<"->"<<ans[i];
}
}
cout<<" "<<send_back<<endl;
return 0;
}