PAT甲级-1018 Public Bike Management

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;
}