倒水问题
Sample Input
2
2 3 4 2
96 97 199 62
Sample Output
2 2
9859 62
#include<iostream>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int MAX = 201;
int cap[3],ans[MAX],vis[MAX][MAX];//ans[i]保存目标为 i的最少水量。 vis保存前两杯水的存量状态(第三杯水就确定了)
struct Node{
int v[3],dist;
bool operator < (const Node& n) const{//使用优先队列,输出使用水最少的,需要重载<。
return this -> dist > n.dist;
}
};
void bfs(int a,int b,int c,int d){
int i,j,t,sum;
Node s;
memset(vis,0,sizeof(vis));
memset(ans,-1,sizeof(ans));
cap[0] = a;
cap[1] = b;
cap[2] = c;
s.v[0] = 0;
s.v[1] = 0;
s.v[2] = c;
s.dist = 0;
priority_queue<Node> q;
q.push(s);
vis[0][0] = 1;
while(!q.empty()){
Node f = q.top();
q.pop();
for(i = 0;i < 3; ++i){//更新状态
t = f.v[i];
if(ans[t] == -1 || ans[t] > f.dist)
ans[t] = f.dist;
}
if(ans[d] != -1)
break;//不会证明。
for(i = 0;i < 3; ++i){
for(j = 0;j < 3; ++j){
if(i != j){
if((f.v[i] != 0) && (f.v[j] != cap[j])){
sum = min(cap[j],f.v[i] + f.v[j]) - f.v[j];//i -> j 倒水
Node n;
memcpy(&n,&f,sizeof(Node));
n.v[i] = f.v[i] - sum;
n.v[j] = f.v[j] + sum;
n.dist = f.dist + sum;
if(!vis[n.v[0]][n.v[1]]){
vis[n.v[0]][n.v[1]] = 1;
q.push(n);
}
}
}
}
}
}
while(d){//使用ans数组的作用来喽。
if(ans[d] != -1){
printf("%d %d\n",ans[d],d);
return;
}
--d;
}
}
int main(){
int n,a,b,c,d;
cin >> n;
while(n--){
scanf("%d%d%d%d",&a,&b,&c,&d);
bfs(a,b,c,d);
}
return 0;
}
代码来自紫书。