分析:取出水量最少的节点进行扩展,这样的程序只需要把队列queue换成优先队列priority_queue,其他部分的代码不变,下面的代码把状态(三元组)和dist合起来定义为一个Node类型,是一种常见的写法,如果要打印路径,需要把访问过的所有节点放在一个nodes数组中,然后在Node中加一个变量fa,表示父节点在nodes数组中的下标,而在队列中只存在节点在nodes数组中的下标而非节点本身,如果内存充足,也可以直接在Node中用一个vector保存路径,省去fa顺着往回找的麻烦。
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
struct Node
{
int v[3], dist;
bool operator < (const Node& rhs) const
{
return dist > rhs.dist;
}
};
const int maxn=205;
int vis[maxn][maxn],cap[maxn],ans[maxn];
void update_ans(const Node& u)
{
for(int i=0;i<3;i++){
int d=u.v[i];
if(ans[d]<0||u.dist<ans[d]) ans[d]=u.dist;
}
}
void solve(int a,int b,int c,int d)
{
cap[0]=a;cap[1]=b;cap[2]=c;
memset(vis,0,sizeof(vis));
memset(ans,-1,sizeof(ans));
priority_queue<Node> q;
Node start;
start.dist=0;
start.v[0]=0;start.v[1]=0;start.v[2]=c;
q.push(start);
vis[0][0]=1;
while(!q.empty()){
Node u = q.top(); q.pop();
update_ans(u);
if(ans[d] >= 0) break;
for(int i=0;i<3;i++)
for(int j=0;j<3;j++ ) if(i!=j){
if(u.v[i]==0 || u.v[j]==cap[j]) continue;
int amount = min(cap[j],u.v[i]+u.v[j]) - u.v[j];
Node u2;
memcpy(&u2, &u, sizeof(u));
u2.dist = u.dist + amount;
u2.v[i] -= amount;
u2.v[j] += amount;
if(!vis[u2.v[0]][u2.v[1]]){
vis[u2.v[0]][u2.v[1]]=1;
q.push(u2);
}
}
}
while(d >=0)
{
if(ans[d] >=0 ){
printf("%d %d\n",ans[d],d);
return;
}
d--;
}
}
int main()
{
int T,a,b,c,d;
scanf("%d",&T);
while(T--){
scanf("%d%d%d%d",&a,&b,&c,&d);
solve(a,b,c,d);
}
return 0;
}