蘑菇街春招 - 编程题第二题 - 隐式图搜索 - 四个水杯

#include <bits/stdc++.h>
using namespace std;
struct CUP
{
int x, y;//x表示当前杯子内水的体积
//y表示这个杯子的容量
};
struct Node
{
CUP a[4];//4个杯子的状态
int step;
} st, ENDDD;
bool flag[111][111][111][111];//标记
int dir[12][2] = {//dir[i][j]表示杯子i向杯子j倒水
0, 1,
0, 2,
0, 3,

1, 0,
1, 2,
1, 3,

2, 0,
2, 1,
2, 3,

3, 0,
3, 1,
3, 2,
};

int BFS()
{

queue<Node> que;
que.push(st);
memset(flag, false, sizeof(flag));
flag[10][0][0][0] = true;

while (!que.empty())
{
Node pre = que.front();
que.pop();

if (pre.a[0].x == ENDDD.a[0].x && pre.a[1].x == ENDDD.a[1].x && pre.a[2].x == ENDDD.a[2].x && pre.a[3].x == ENDDD.a[3].x)
return pre.step;

for (int i = 0; i < 12; i++)
{
int a = dir[i][0], b = dir[i][1];

Node tmp = pre; // 要用一个中间变量，不然状态会被改变

if (tmp.a[a].x == 0 || tmp.a[b].x == tmp.a[b].y)
continue;

if (tmp.a[a].x >= tmp.a[b].y - tmp.a[b].x)//能倒满
{
tmp.a[a].x -= (tmp.a[b].y - tmp.a[b].x);//注意这两条语句的顺序，很重要
tmp.a[b].x = tmp.a[b].y;
}
else if (tmp.a[a].x + tmp.a[b].x <= tmp.a[b].y)// 倒空
{
tmp.a[b].x += tmp.a[a].x;
tmp.a[a].x = 0;
}

if (flag[tmp.a[0].x][tmp.a[1].x][tmp.a[2].x][tmp.a[3].x] == false)
{
flag[tmp.a[0].x][tmp.a[1].x][tmp.a[2].x][tmp.a[3].x] = true;
tmp.step = pre.step + 1;
que.push(tmp);
}
}
}
return 0;
}

int main()
{

int a, b, c, d;
while (~scanf("%d%d%d%d", &a, &b, &c, &d))
{
if (a + b + c + d != 10)
{
printf("-1\n");
continue;
}
st.a[0].x = 10;//初始化
st.a[0].y = 10;

st.a[1].x = 0;
st.a[1].y = 6;

st.a[2].x = 0;
st.a[2].y = 5;

st.a[3].x = 0;
st.a[3].y = 4;

st.step = 0;

ENDDD.a[0].x = a;
ENDDD.a[0].y = 10;

ENDDD.a[1].x = b;
ENDDD.a[1].y = 6;

ENDDD.a[2].x = c;
ENDDD.a[2].y = 5;

ENDDD.a[3].x = d;
ENDDD.a[3].y = 4;

ENDDD.step = 0;
int ans = BFS();
printf("%d\n", ans);
}
return 0;
}