NYOJ 21 三个水杯(bfs模拟)

#include <iostream>
#include <queue>
#include <cstring>
using namespace std;
struct Node{
	int cup[3];
	int step;
}Now,Next;
int S[3],E[3],vis[105][105][105];        // 标记数组开大点
int bfs(){
	Now.cup[0] = S[0];
	Now.cup[1] = 0;
	Now.cup[2] = 0;
	Now.step = 0;
	vis[Now.cup[0]][Now.cup[1]][Now.cup[2]] = 1;         // 把初始状态标记
	queue<Node> q;
	q.push(Now);
	while(!q.empty()){
		Now = q.front();
		q.pop();
		if(Now.cup[0]==E[0]&&Now.cup[1]==E[1]&&Now.cup[2]==E[2]){
			return Now.step;
		}
		for(int i=0;i<3;i++){                //  用两个for循环模拟  从i->j倒水的过程
			for(int j=0;j<3;j++){
				if(i!=j){
					int water = S[j] - Now.cup[j];                 // water为这个杯子还能装多少水
					if(Now.cup[i]<water){                          // 判断这个杯子是否能倒满
						water = Now.cup[i];
					}
					Next = Now;                     // 及时初始化Next 否则只有一个杯子更新数据
					Next.cup[i] = Now.cup[i] - water;              // 从i被子里倒水
					Next.cup[j] = Now.cup[j] + water;							 // j杯子里加入这个多水
					if(vis[Next.cup[0]][Next.cup[1]][Next.cup[2]]==0){
						vis[Next.cup[0]][Next.cup[1]][Next.cup[2]] = 1;
						Next.step = Now.step + 1;
						q.push(Next);
					}
				}
			}
		}
	}
	return -1;
}
int main()
{
	int n;
	cin>>n;
	while(n--){
		memset(vis,0,sizeof(vis));
		cin>>S[0]>>S[1]>>S[2]>>E[0]>>E[1]>>E[2];
		if(E[0]+E[1]+E[2]!=S[0]){            // 剪枝
			cout<<"-1"<<endl;
		}
		else{
			cout<<bfs()<<endl;
		}
	}
	return 0;
}
/***
   [来源] NYOJ 21
	 [题目] 三个水杯
	 [思路]
	    用结构体存一个cup数组表示三个水杯，然后根据bfs搜索，用两个for循环模拟倒水的过程，
			剩下就是bfs过程，关键在于模拟倒水过程，详细看代码注释。
	 [输入]
	    2
			6 3 1
			4 1 1
			9 3 2
			7 1 1
	 [输出]
	    3
			-1
*/