倒水问题——DFS

目录

1.题目描述

2.题意

3.解题思路

4.代码实现

---

1.题目描述

题目来源：上海交通大学考研机试题

有三个杯子，容量分别为 A,B,C。

初始时，C 杯装满了水，而 A,B 杯都是空的。

现在在保证不会有漏水的情况下进行若干次如下操作：

将一个杯子 x 中的水倒到另一个杯子 y 中，当 x 空了或者 y 满了时就停止（满足其中一个条件才停下）。

请问，在操作全部结束后，C 中的水量有多少种可能性。

输入格式

输入包含多组测试数据。

每组数据占一行，包含三个整数 A,B,C。

输出格式

每组数据输出一个结果，占一行。

数据范围

0≤A,B,C≤4000
每个输入最多包含 100 组数据。

输入样例：

0 5 5
2 2 4

输出样例：

2
3

2.题意

题意：给你三个有固定容量的杯子，其中只有杯子C被装满了水。你可以将一个杯子的水倒到另外一个杯子里，当另外一个杯子的水满了，或倒水的杯子的水倒光了则停止。问通过多次这样的操作，杯子C中水量的总共有多少种存在的可能。

3.解题思路

思路：初始时只有杯子C中装有水，我们要用DFS来解决该问题，可以将当前这个三个杯子中的水量当成一种状态，同时记录当前杯子C的水量。每一次我们可以模拟三个杯子之间倒水的选择，若这一种选择改变这三个杯子中水量的状态没有被搜索过，则记录该状态，并以这种状态再次进行选择，同时记录现在状态中杯子C中的水量，直到所有的状态都搜索过。最后输出杯子C中水量的状态数。

4.代码实现

c[3]：存储三个杯子的容量

s：存储三个杯子水量的状态

cs：存储杯子C水量的状态

代码思路：因为是多组样例输入，所以先将s和cs初始化，然后将当前三个杯子的水量存入一个数组，然后以这个数组为dfs的参数，开始dfs。在dfs中，先将当前三个杯子水量的状态通过get_LL转化为一个整数然后存入s中，再将当前杯子C水量的状态存入cs中。然后开始模拟倒水，倒水的选择共有6种，即0倒1,0倒2,1倒0,1倒2,2倒0,2倒1，若倒水后三个杯子水量的状态没有被搜索过，则以该种状态再进行搜索，最后输出搜到的杯子C的水量的状态数。

#include<iostream>
#include<unordered_set>
#include<cstring>

using namespace std;

typedef long long LL;
int c[3];
LL B = 10000;
unordered_set<LL> s;
unordered_set<int> cs;

LL get_LL(int w[]) { /// 将三个的杯的子水量转化为一个整数
    return w[0] * B * B + w[1] * B + w[2];
}

void pure(int a[], int i, int j) {
    int t = min(a[i], c[j] - a[j]);
    a[i] -= t, a[j] += t;
}

void dfs(int w[]) {
    cs.insert(w[2]); ///存储当前杯子C的水量状态
    s.insert(get_LL(w)); ///存储当前三个杯子的水量状态

    int a[3];
    /// 模拟三个杯子之间的倒水可能，i倒j
    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 3; j++) {
            if (i != j) {///非自己给自己倒水
                 ///将当前三个杯子水量的状态赋给a数组
                memcpy(a, w, sizeof a); 
                pure(a, i, j); ///将i的水倒给j
                if (!s.count(get_LL(a))) { ///若当前状态未被搜过则进行进一步搜索
                    dfs(a);
                }
            }
        }
    }
}

int main() {
    while (cin >> c[0] >> c[1] >> c[2]) {
        /// 初始化
        s.clear();
        cs.clear();

        int w[3] = { 0,0,c[2] };///三个杯子刚开始的水量状态
        dfs(w);
        cout << cs.size() << endl;
    }
    return 0;
}