(BFS搜索)【HDU 1495】非常可乐

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1495

最开始没太懂题意,以为另外两个瓶子里的值相等就行了。。搜了题解才知道要用BFS,记录每一步的状态,一次一次把每一个有可乐的杯子倒到另外的杯子里去,直到所有的可乐被平分在两个容器里(可乐瓶+一个杯子也可以)。

需要注意的就是多组样例,每一次需要把队列清空,所以最好就在bfs函数里创建队列。


#include<map>
#include<set>
#include<cmath>
#include<stack>
#include<queue>
#include<deque>
#include<string>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
#define PI 3.1415926
#define INF 0x3f3f3f3f
using namespace std;
const int MAXN = 100+10;

int a[4],ans,vis[MAXN][MAXN][MAXN];

struct node
{
    int n[4],cur;
}tmp,p;

void bfs()
{
    queue<node> que;
    tmp.n[1] = a[1];
    tmp.n[2] = tmp.n[3] = tmp.cur = 0;
    que.push(tmp);
    while(!que.empty())
    {
        tmp = que.front();
        que.pop();
        for(int i = 1; i <= 3; ++i)
        {
            if(tmp.n[i] <= 0)   continue;
            for(int j = 1; j <= 3; ++j)
            {
                if(j == i)  continue;
                p = tmp;
                if(p.n[i] < a[j]-p.n[j]) //i不能倒满j
                {
                    p.n[j] += p.n[i];
                    p.n[i] = 0;
                }
                else//i的可乐能装满j
                {
                    p.n[i] -= a[j]-p.n[j];
                    p.n[j] = a[j];
                }
                if(!vis[p.n[1]][p.n[2]][p.n[3]])
                {
                    vis[p.n[1]][p.n[2]][p.n[3]] = 1;
                    ++p.cur;
                    if(((p.n[1]==p.n[2] || p.n[1]==p.n[3]) && p.n[1]==a[1]/2) || (p.n[2]==p.n[3] && p.n[2]==a[1]/2)) //找到解的状态
                    {
                        cout<<p.cur<<'\n';
                        return;
                    }
                    que.push(p);
                }
            }
        }
    }
    cout<<"NO\n";
}
int main(void)
{
    while(cin>>a[1]>>a[2]>>a[3],a[1]&&a[2]&&a[3])
    {
        if(a[1]%2)  {cout<<"NO\n"; continue;}
        memset(vis,0,sizeof vis);
        vis[a[1]][a[2]][a[3]] = 1;
        bfs();
    }
}

猜你喜欢

转载自blog.csdn.net/suiguia/article/details/80058508