LeetCode 365. 水壶问题--BFS模拟

365. 水壶问题

有两个容量分别为 x升 和 y升 的水壶以及无限多的水。请判断能否通过使用这两个水壶，从而可以得到恰好 z升 的水？

如果可以，最后请用以上水壶中的一或两个来盛放取得的 z升 水。

你允许：

装满任意一个水壶
清空任意一个水壶
从一个水壶向另外一个水壶倒水，直到装满或者倒空

示例 1: (From the famous “Die Hard” example)

输入: x = 3, y = 5, z = 4
输出: True

示例 2:

输入: x = 2, y = 6, z = 5
输出: False

题解：

直接模拟吧，因为数据量在1e6内（英文题目界面说明了），所以直接定义bool vis[1000100]标记。

AC代码：

class Solution {
public:
    bool vis[1000100];
    queue<int>q;
    bool bfs(int X,int Y,int Z)
    {
        while(!q.empty())q.pop();
        memset(vis,0,sizeof(vis));
        if(X==Z||Y==Z||(X+Y)==Z)return true;
        if(X<Y)swap(X,Y);
        q.push(X-Y);
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            for(int i=0;i<4;i++)
            {
                int v;
                if(i==0)
                {
                    if(Y-u<0)continue;
                    v=X-(Y-u);
                    if(v<0)continue;
                    //cout<<v<<endl;
                    if(vis[v])continue;
                    vis[v]=true;
                    q.push(v);
                }
                else if(i==1)
                {
                    if(X-u<0)continue;
                    v=Y-(X-u);
                    if(v<0)continue;
                    if(vis[v])continue;
                    vis[v]=true;
                    q.push(v);
                }
                else if(i==2)
                {
                    v=min(X,Y+u);
                    if(vis[v])continue;
                    vis[v]=true;
                    q.push(v);
                }
                if(v==Z||(X+v)==Z||(Y+v)==Z)return true;
            }
        }
        return false;
    }
    bool canMeasureWater(int x, int y, int z) {
        if(x+y<z)return false;
        return bfs(x,y,z);
    }
};