Week2作业B题 倒水问题

题目：

给你两个容器，容量分别为A， B ，问是否能够经过有限的步骤倒水, 得到容量为C 的水。
“fill A” 表示倒满A杯，"empty A"表示倒空A杯，“pour A B” 表示把A的水倒到B杯并且把B杯倒满或A倒空。

输入：
输入包含多组数据。每组数据输入 A, B, C 数据范围 0 < A <= B 、C <= B <=1000 、A和B互质。

输出：
你的程序的输出将由一系列的指令组成。这些输出行将导致任何一个罐子正好包含C单位的水。每组数据的最后一行输出应该是“success”。输出行从第1列开始，不应该有空行或任何尾随空格。

样例：
Input:
2 7 5
2 7 4
Output:
fill B
pour B A
success
fill A
pour A B
fill A
pour A B
success

思路：

我们把一个状态看作是一个节点。初始状态为起点，目标状态为终点。
状态的变化无非四种：
1.A倒入B
2.B导入A
3.A倒空/倒满
4.B倒空/倒满
把变化后的状态看作是可访问的另一个邻接点，之后运用BFS的思路方法进行实现，看是否能到达目标状态。
在此过程中需要记录某状态是否已经访问过，以及每个状态的前一个状态(本代码使用map)。

代码：

#include <stdio.h>
#include <map>
#include <queue>
using namespace std;

struct Status 
{
	int a, b;//a中的水量，b中的水量 
	int op;//用于记录操作 
	bool operator<(const Status &s) const //如果不重载将无法通过编译 
	{
		return a!=s.a ? a<s.a : b<s.b;
	}
};
queue<Status> Q;
map<Status, Status> from;//用于记录前一状态 

int la=0;//标记是否到达终点 
void print_op(int t){//把每种操作进行编号 
	if(t==0) printf("fill A\n");
	if(t==1) printf("empty A\n");
	if(t==2) printf("fill B\n");
	if(t==3) printf("empty B\n");
	if(t==4) printf("pour A B\n");
	if(t==5) printf("pour B A\n");
}

/* 递归输出方案 */
void print(Status &p,int t) 
{
    if ( from.find(p) == from.end() || (p.a == 0&&p.b==0) )
    {
    	//printf("<%d,%d>",  p.a, p.b);
    	print_op(p.op);
        return;
    }
    print(from[p],1); //递归
    //printf("-><%d,%d>",  p.a, p.b);
    if(t!=0)//不采用此写法会导致多输出一行 
        print_op(p.op);
}

void refresh(Status &s, Status &t) 
{
    if ( from.find(t) == from.end() ) //判断是否已经访问过此状态 
    { 
        from[t] = s;
        Q.push(t);
    }
}

void bfs(int A, int B, int C) 
{
	// 起点, 两杯水都空
	Status s,t; s.a=0; s.b=0;
	s.op=-1,t.op=-1;
	Q.push(s);

    while (!Q.empty()) 
    {
    	// 取队首
        s = Q.front(); Q.pop();
        // 特判到达终点
        if (s.a == C || s.b == C) {
            print(s,0); // 输出方案
            return;
        }

        // 倒空 a 杯的水
        if (s.a > 0) {
            t.a = 0;  // 倒空
            t.b = s.b;// b 杯不变
            s.op=1,t.op=1;
            refresh(s, t);
        }

        // 同理，倒空 b 杯的水
        if (s.b > 0) {
            t.b = 0;  // 倒空
            t.a = s.a;// a 杯不变
            s.op=3,t.op=3;
            refresh(s, t);
        }

        // a 杯未满，续满 a 杯
        if (s.a < A) 
        {
        	// 续满 a 杯
        	t.a = A;  
        	t.b = s.b;
        	s.op=0,t.op=0;
            refresh(s, t);

            // 考虑倒入
            if (s.b != 0) 
            {
                if (s.a + s.b <= A) 
                {
                    t.a = s.a + s.b;
                    t.b = 0;
                    s.op=5,t.op=5;
           			refresh(s, t);
                } else 
                {
                    t.a = A;
                    t.b = s.a + s.b - A;
                    s.op=5,t.op=5;
            		refresh(s, t);
                }
            }
        }

        // 同理，b 杯未满，续满 b 杯
        if (s.b < B) 
        {
            t.a = s.a;
            t.b = B;
            s.op=2,t.op=2;
            refresh(s, t);
            if (s.a != 0) 
            {
                if (s.a + s.b <= B) 
                {
                    t.a = 0;
                    t.b = s.a + s.b;
                    s.op=4,t.op=4;
                    refresh(s, t);
                } else 
                {
                    t.a = s.a + s.b - B;
                    t.b = B;
                    s.op=4,t.op=4;
                    refresh(s, t);
                }
            }
        }
    }
    printf("-1\n");
}
int main() 
{
	
    int a, b, c; 
    while(scanf("%d%d%d", &a, &b, &c)!=EOF){
    	 bfs(a, b, c);
    	 printf("success\n");
	}
    
    return 0;
}

总结：

本题中使用map，既可以避免重复访问相同状态，又可以记录前一状态，实现回溯。

本代码是在学长代码的基础上修改的。参考链接：https://paste.ubuntu.com/p/VpgG59MPXS/