NYOJ21-三个水杯

三个水杯

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 4

描述

给出三个水杯，大小不一，并且只有最大的水杯的水是装满的，其余两个为空杯子。三个水杯之间相互倒水，并且水杯没有标识，只能根据给出的水杯体积来计算。现在要求你写出一个程序，使其输出使初始状态到达目标状态的最少次数。

输入

第一行一个整数N(0<N<50)表示N组测试数据
接下来每组测试数据有两行，第一行给出三个整数V1 V2 V3 (V1>V2>V3 V1<100 V3>0)表示三个水杯的体积。
第二行给出三个整数E1 E2 E3 （体积小于等于相应水杯体积）表示我们需要的最终状态

输出

每行输出相应测试数据最少的倒水次数。如果达不到目标状态输出-1

样例输入

2
6 3 1
4 1 1
9 3 2
7 1 1

样例输出

3
-1

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;

struct DateType
{
    int state[3];
    int step;
};
int vis[100][100][100],Cup[3],Target[3];
void PowerWater(int i,int j,DateType& newnode)//注意：此处第三个参数为引用
{
    int water=Cup[j]-newnode.state[j];
    if(newnode.state[i]>water)
    {
        newnode.state[i]-=water;
        newnode.state[j]+=water;
    }
    else
    {
        newnode.state[j]+=newnode.state[i];
        newnode.state[i]=0;
    }
    newnode.step+=1;
}
int IsAchieveTarget(DateType node)
{
    for(int i=0;i<3;i++)
    {
        if(node.state[i]!=Target[i])
            return 0;
    }
    return 1;
}
int BFS()
{
    memset(vis,0,sizeof(vis));
    int i,j,k;
    DateType initial;
    initial.state[0]=Cup[0];
    initial.state[1]=initial.state[2]=0;
    initial.step=0;
    queue<DateType> qstate;
    qstate.push(initial);
    vis[initial.state[0]][initial.state[1]][initial.state[2]]=1;
    while(!qstate.empty())
    {
        DateType node=qstate.front();
        qstate.pop();
        if(IsAchieveTarget(node))
        {
            return node.step;
        }
        for(i=0;i<3;i++)
        {
            for(j=1;j<3;j++)
            {
                k=(i+j)%3;
                if(node.state[i]&&node.state[k]<Cup[k])//第i个杯子非空且第k个杯子非满
                {
                    DateType newnode=node;
                    PowerWater(i,k,newnode);//第i个杯子想第k个杯子倒水
                    if(!vis[newnode.state[0]][newnode.state[1]][newnode.state[2]])
                    {
                        qstate.push(newnode);
                        vis[newnode.state[0]][newnode.state[1]][newnode.state[2]]=1;
                    }
                }
            }
        }
    }
    return -1;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>Cup[0]>>Cup[1]>>Cup[2];
        cin>>Target[0]>>Target[1]>>Target[2];
        cout<<BFS()<<endl;
    }
    return 0;
}

参考别人的代码后，自己写了一遍，因为一天判断条件写错了，浪费了一个多小时。痛心啊。。。