[容易] bfs状态搜索

九度oj1457

题目描述：

大家一定觉的运动以后喝可乐是一件很惬意的事情，但是seeyou却不这么认为。因为每次当seeyou买了可乐以后，阿牛就要求和seeyou一起分享这一瓶可乐，而且一定要喝的和seeyou一样多。但seeyou的手中只有两个杯子，它们的容量分别是N 毫升和M 毫升 可乐的体积为S （S<101）毫升(正好装满一瓶) ，它们三个之间可以相互倒可乐 (都是没有刻度的，且 S==N+M，101＞S＞0，N＞0，M＞0) 。聪明的ACMER你们说他们能平分吗？如果能请输出倒可乐的最少的次数，如果不能输出"NO"。

输入：

三个整数 : S 可乐的体积 , N 和 M是两个杯子的容量，以"0 0 0"结束。

输出：

如果能平分的话请输出最少要倒的次数，否则输出"NO"。

样例输入：

7 4 3
4 1 3
0 0 0

样例输出：

NO
3

思路：

首先将题目建模为状态以及状态转换，然后又是求最小步数，自然想到BFS，可以用BFS来做一个状态搜索。

具体代码：

#include <iostream>
#include <queue>
#include <cstring>

using namespace std;
struct state
{
    int vol[3];
    int step;
    state(int s=0,int n=0,int m=0,int st=0)
    {
        vol[0]=s;
        vol[1]=n;
        vol[2]=m;
        step=st;
    }
};
int S,N,M;
int getMaxVol(int id)
{
    switch(id)
    {
    case 0:
        return S;
    case 1:
        return N;
    case 2:
        return M;
    }
}
const int maxn=100+5;
bool visited[maxn][maxn][maxn];
void pour(const state& ori,int a,int b,state& ans)
{
    ans=ori;
    int bm=getMaxVol(b);
    int bl=bm-ori.vol[b];
    int al=ori.vol[a]-bl;
    if(al<=0)
    {
        ans.vol[a]=0;
        ans.vol[b]=ori.vol[a]+ori.vol[b];
    }
    else
    {
        ans.vol[a]=al;
        ans.vol[b]=bm;
    }
    ans.step=ori.step+1;
}
int bfs()
{
    memset(visited,0,sizeof visited);
    int final=S/2;
    queue<state> q;
    q.push(state(S,0,0,0));
    while(!q.empty())
    {
        state ori=q.front();q.pop();
        visited[ori.vol[0]][ori.vol[1]][ori.vol[2]]=true;
        for(int i=0;i<3;++i)
        {
            for(int j=0;j<3;++j)
            {
                if(i==j) continue;
                state ans;
                pour(ori,i,j,ans);
                if(ans.vol[0]==final&&ans.vol[1]==final||
                        ans.vol[1]==final&&ans.vol[2]==final||ans.vol[0]==final&&ans.vol[2]==final)
                {
                    return ans.step;
                }
                if(!visited[ans.vol[0]][ans.vol[1]][ans.vol[2]])
                    q.push(ans);

            }
        }
    }
    return -1;
}

int main()
{
    for(cin>>S>>N>>M;S!=0;cin>>S>>N>>M)
    {
        if(S%2==1)
        {
            cout<<"NO"<<endl;
            continue;
        }
        int st=bfs();
        if(st==-1)
            cout<<"NO"<<endl;
        else
            cout<<st<<endl;
    }
    return 0;
}