Fill UVA - 10603

UVA-10603

题目：给容量为a，b，c的杯子，开始只有第三个杯子装满水，另外两个为空。问最少要倒多少升水才能让其中某一个杯子中有水d升？如果没有办法恰好做到d升，就让某一个杯子里的水为D升，其中D<d并且尽量接近d。(1<=a,b,c,d<=200)。输出最少的倒水量和目标水量。

分析：用BFS。由于目标是最少倒水量，不是步骤少的优先，于是考虑用优先队列，让倒水量少的状态先出列进行拓展。

        由于总水量一定，只要知道其中两个水杯的水量，即可推出第三个水杯的水量。于是用vis[i][j]来判重，避免无效结点的拓展。

#include<cstdio>
#include<cmath>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<sstream>
#include<map>
#include<stack>
#include<queue>
using namespace std;
int T,a,b,c,need;
int vis[205][205],ans[201],s[4];
struct Node{
    int d,v[4];
    bool operator < (const Node& n1) const {
        return d > n1.d;
    }
};
/*
 优先队列：
    重载小于号运算符，如果 d>n1.d为true，则Node $ < n1为true，于是n1先出队列，于是倒水量小的优先出队
*/
void update_ans(Node t){//更新ans的值
    for(int i = 1;i<=3;i++){
        if(ans[t.v[i]]==-1||ans[t.v[i]] > t.d) ans[t.v[i]] = t.d;
    }//ans[i]表示到水量i的最少倒水量数
}
void bfs(int a,int b,int c,int d){
    memset(vis, 0, sizeof(vis)); //判重数组
    memset(ans, -1, sizeof(ans));
    priority_queue<Node>q;
    Node ori;
    ori.v[1] = 0;ori.v[2] = 0;ori.v[3] = c;ori.d = 0;
    vis[0][0] = 1;
    q.push(ori);
    while(!q.empty()){
        Node u = q.top();q.pop();
        update_ans(u);
        if(ans[d]>=0)break; //找到答案
        for(int i = 1;i<=3;i++){ //把i号杯中的水给j号杯
            for(int j = 1;j<=3;j++){
                if(i!=j){
                    int z = 0;
                    if(i+j==3) z = 3;
                    if(i+j==4) z = 2;
                    if(i+j==5) z = 1;
                    int start = u.v[i],end = u.v[j],moveto;
                    Node new_node;
                    if(start+end<=s[j]){
                        moveto = start;
                        new_node.v[i] = 0;
                        new_node.v[j] = start+end;
                        new_node.v[z] = u.v[z];
                        new_node.d = u.d+moveto;
                        if(!vis[new_node.v[1]][new_node.v[2]]){vis[new_node.v[1]][new_node.v[2]] = 1;q.push(new_node);}
                    }else{
                        moveto = s[j] - end;
                        new_node.v[i] = start - moveto;
                        new_node.v[j] = s[j];
                        new_node.v[z] = u.v[z];
                        new_node.d = u.d+moveto;
                        if(!vis[new_node.v[1]][new_node.v[2]]){vis[new_node.v[1]][new_node.v[2]] = 1;q.push(new_node);}
                    }
                }
            }
        }
    }
    while(d>=0){//输出答案
        if(ans[d]>=0){
            printf("%d %d\n",ans[d],d);
            return;
        }
        d--;
    }
}
int main(){
    cin>>T;
    while(T--){
        scanf("%d%d%d%d",&s[1],&s[2],&s[3],&need);
        bfs(s[1],s[2],s[3],need);
    }
    return 0;
}