UVa 10603 倒水问题

https://vjudge.net/problem/UVA-10603

题意：三个杯子，倒水问题。找出最少倒水量。

思路：路径寻找问题。不难，暴力枚举。

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

int a, b, c, d;
int vis[205][205]; //访问变量，因为只有三个杯子，所以记录前两个杯子的状态即可
int cap[3];
int ans[205];

struct Node
{
    int v[3], dist;
    bool operator <(const Node& rhs) const //重载小于号
    {
        return dist>rhs.dist;
    }
};

void update(Node u)  //每次有新状态时，记录形成d升水时需要的最少倒水量
{
    for (int i = 0; i < 3; i++)
    {
        int x = u.v[i];
        if (ans[x] < 0 || u.dist < ans[x])    ans[x] = u.dist;
    }
}

void bfs()
{
    cap[0] = a, cap[1] = b, cap[2] = c; //记录三个杯子的容量
    memset(vis, 0, sizeof(vis));
    memset(ans, -1, sizeof(ans));
    priority_queue<Node> q;
    Node start;
    start.v[0] = 0; start.v[1] = 0; start.v[2] = c; start.dist = 0;
    q.push(start);
    vis[0][0] = 1; //设置初始状态为已访问
    while (!q.empty())
    {
        Node p = q.top();
        q.pop();

        update(p);
        if (ans[d] >= 0) break;  //找到目标状态，退出循环

        for (int i = 0; i < 3; i++)
        {
            for (int j = 0; j < 3; j++)
            {

                if (i != j && p.v[i]!=0 && p.v[j]<cap[j]) //不能给自己倒水并且第i个杯子有水和第j个杯子未满
                {
                    Node u=p; //建立新状态
                    int r = cap[j] - p.v[j];  //计算第j个杯子空余水量
                    if (p.v[i] >= r) //如果第i个杯子水量充足
                    {
                        u.dist = p.dist + r;  //更新后的倒水量
                        u.v[j] = p.v[j] + r; //更新后第j个杯子水量
                        u.v[i] = p.v[i] - r; //更新后第i个杯子水量
                    }
                    else //如果第i个杯子水量不足
                    {
                        u.dist = p.dist + p.v[i];
                        u.v[j] = p.v[j] + p.v[i];
                        u.v[i] = 0;
                    }
                    if(!vis[u.v[0]][u.v[1]]) //如果该状态没出现过
                    {
                       vis[u.v[0]][u.v[1]] = 1;
                       q.push(u);
                    }
                }


            }
        }

    }

    while (d >= 0)
    {
        if (ans[d] >= 0)
        {
            cout << ans[d] << " " << d << endl;
            return;        
        }
        d--; //如果无法做到d升，则检验d--，找到最接近d升
    }
}

int main()
{
    //freopen("D:\txt.txt","r", stdin);
    int n;
    cin >> n;
    while (n--)
    {
        cin >> a >> b >> c >> d;
        bfs();
    }
    return 0;
}