【Dijkstra+DFS】
总共存在三个标尺——最短路径、PBMC派出去的自行车数、PBMC收回的自行车数,同时还需要求解路径。可以说是非常麻烦了qaq
坑点:所有车站在调整过程中必须一步到位,不能用路径靠后的结点来调整路径靠前的结点。(然原题目里好像并没有说明这件事,不过这个坑点只值5分)
非常好的一组测试点(源自《算法笔记》):
//Input
10 4 4 5
4 8 9 0
0 1 1
1 2 1
1 3 2
2 3 1
3 4 1
//Output
1 0->1->2->3->4 2
附上AC代码~不过写的可能有点啰嗦- -原谅我这个小菜鸡
#include <bits/stdc++.h>
using namespace std;
const int INF=10000000;
struct node
{
int v,w;
};
vector<node> v[512];
int bike[512];
vector<int> pre[512];
int d[512];
bool vst[512];
vector<int> ans,tmp;
int in=INF,out=INF,n,c;
void Dijkstra(int s)
{
memset(vst,false,sizeof(vst));
fill(d,d+512,INF);
d[0]=0;
for(int i=0;i<=n;i++)
{
int u=-1,min=INF;
for(int j=0;j<=n;j++)
if(!vst[j]&&d[j]<min)
{
u=j;min=d[j];
}
if(u==-1) return;
vst[u]=true;
for(int j=0;j<v[u].size();j++)
{
if(!vst[v[u][j].v]&&d[u]+v[u][j].w<d[v[u][j].v])
{
d[v[u][j].v]=d[u]+v[u][j].w;
pre[v[u][j].v].clear();
pre[v[u][j].v].push_back(u);
}
else if(!vst[v[u][j].v]&&d[u]+v[u][j].w==d[v[u][j].v])
pre[v[u][j].v].push_back(u);
}
}
}
void DFS(int s)
{
if(s==0)
{
int i=0,o=0,TEMP;
tmp.push_back(0);
for(int j=tmp.size()-2;j>=0;j--)
{
int db=bike[tmp[j]]-c/2;
if(db<0) //need bikes
{
db=-db;
TEMP=min(db,i);
i-=TEMP;
db-=TEMP;
o+=db;
}
else i+=db;
}
if(o<out)
{
out=o;in=i;ans=tmp;
}
else if(o==out&&i<in)
{
in=i;ans=tmp;
}
tmp.pop_back();
return;
}
else
{
tmp.push_back(s);
for(int i=0;i<pre[s].size();i++)
DFS(pre[s][i]);
tmp.pop_back();
}
}
int main(void)
{
int p,m;
scanf("%d%d%d%d",&c,&n,&p,&m);
for(int i=1;i<=n;i++)
scanf("%d",&bike[i]);
for(int i=0;i<m;i++)
{
int si,sj,tij;
node n1,n2;
scanf("%d%d%d",&si,&sj,&tij);
n1.v=si;n1.w=tij;
n2.v=sj;n2.w=tij;
v[si].push_back(n2);
v[sj].push_back(n1);
}
Dijkstra(p);
DFS(p);
printf("%d ",out);
for(int i=ans.size()-1;i>=0;i--)
{
printf("%d",ans[i]);
if(i!=0) printf("->");
}
printf(" %d",in);
}