倒水问题(Fill UVa 10603)
题目分析:
题目要求输出最少的倒水量和操作结束后是否有一个水杯为目的水量(当不能恰好为d时要尽量接近d,输出一个d')
假设在某一时刻的状态为第一个杯子cup[0]中有v[0]升水,第二个cup[1]中有v[1]升水,第三个cup[2]中有v[2]升水,此时的状态就可以直接用结构体Node进行记录,由于题目中1<=a,b,c,d<=200,而每个杯子的状态有0,1,2...x种情况,即x+1种可能,那么状态最大值为201^3=8120601种,又因为状态中有重复的,故不会超过201^2=40401(当两个状态中cup1,cup2的v都相同时,cup3的v也是一样的)。
又因为题目要求的不是倒水次数最少而是倒水量最少,所以用优先队列的数据结构(修改排序方式)每次将最小的倒水量的结点放在堆顶。如果要保存路径可以在Node中加一个vector数组记录,代码暂时没有给出。
(没有看原题的样例,看书上的样例,给出两个例子)
样例:
输入(多组)
2
1 3 6 4
1 12 15 7
输出
3 4
14 7
下面给出代码(参考算法紫书):
//author : Lmry
//time : 2021.5.22
//something else : This is my first blog.
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
struct Node{
int v[3],dist;//dist代表倒水的量
bool operator < (const Node& rhs) const{
return dist>rhs.dist;
}
//优先队列与sort相反,如果dist>rhs.dist,就是从小到大排序
};
const int N = 205;
int vis[N][N],cup[3],ans[N];
void update_ans(const Node&x){
for(int i=0;i<3;i++){
int d=x.v[i];
if(ans[d]<0||x.dist<ans[d])
ans[d]=x.dist;
}
}
void solve(int a,int b,int c,int d){
cup[0]=a,cup[1]=b,cup[2]=c;
memset(vis,0,sizeof(vis));
memset(ans,-1,sizeof(ans));
priority_queue<Node> q;
Node start;
start.v[0]=0,start.v[1]=0,start.v[2]=c;
start.dist=0;
q.push(start);
vis[0][0]=1;
while(!q.empty()){
Node x = q.top();
q.pop();
update_ans(x);
for(int i=0;i<3;i++){
for(int j=0;j<3;j++){
//如果有一个杯子是空的或者有一个是装满的,就不操作
if(x.v[i]==0||x.v[j]==cup[j]) continue;
int amount=min(cup[j],x.v[i]+x.v[j])-x.v[j];//amount是倒满j或者倒空i的倒水量
Node y;//下一个状态
memcpy(&y,&x,sizeof(x));
y.dist=x.dist+amount;
y.v[i]-=amount;
y.v[j]+=amount;
if(!vis[y.v[0]][y.v[1]]){
vis[y.v[0]][y.v[1]]=1;
q.push(y);
}
}
}
}
while(d>=0){
if(ans[d]>=0){
cout<< ans[d] <<" "<< d <<endl;
return;
}
d--;
}
}
int main(){
int T,a,b,c,d;
cin>>T;
while(T--){
cin>>a>>b>>c>>d;
solve(a,b,c,d);
}
return 0;
}
//Thank you for reading.