USACO-Section1.4 Mother's Milk [搜索]

2017-6-9

题目大意

农民约翰有三个容量分别是A,B,C升的桶，A,B,C分别是三个从1到20的整数， 最初，A和B桶都是空的，而C桶是装满牛奶的。有时，农民把牛奶从一个桶倒到 另一个桶中，直到被灌桶装满或原桶空了。当然每一次灌注都是完全的。由于节约， 牛奶不会有丢失
写一个程序去帮助农民找出当A桶是空的时候，C桶中牛奶所剩量的所有可能性。

题解

搜索。
一共有三个杯子，如果某个杯子中有牛奶，那就可以向另外两个杯子中的任意一个倒牛奶。要么倒满被倒的杯子，要么倒空当前的杯子。
当第一个杯子容量为空时，就记录第三个杯子的当前容量。
用一个visit[]数组记录三个杯子的状态，防止无限循环。

代码

/*
ID: zachery1
PROG: milk3
LANG: C++
*/
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>
#include <algorithm>
#define N 3
#define MAXVIS 300000
#define cin fin
#define cout fout
using namespace std;
ifstream fin("milk3.in");
ofstream fout("milk3.out");

vector<int> ans;
int capacity[N], buckets[N];
bool visit[MAXVIS];

void dfs() {
    if (buckets[0] == 0) ans.push_back(buckets[2]);
    for (int i = 0; i < N; i++) {
        if (buckets[i]) { // 杯子不空，可以向另外两个杯子中倒牛奶
            for (int j = 0; j < N; j++) {
                if (j != i) {
                    int m = min(buckets[i], capacity[j]-buckets[j]);
                    // 倒牛奶
                    buckets[i] -= m;
                    buckets[j] += m;
                    int f = buckets[0]*10000 + buckets[1]*100 + buckets[2];
                    if (!visit[f]) {
                        visit[f] = true;
                        dfs();
                    }
                    // 回溯
                    buckets[i] += m;
                    buckets[j] -= m;
                }
            }
        }
    }
}

int main() {
    cin >> capacity[0] >> capacity[1] >> capacity[2];
    memset(buckets, 0, sizeof(buckets));
    memset(visit, false, sizeof(visit));
    buckets[N-1] = capacity[2];
    int f = buckets[0]*10000 + buckets[1]*100 + buckets[2];
    visit[f] = true; // 记录初始状态
    dfs();
    sort(ans.begin(), ans.end());
    for (int i = 0; i < ans.size()-1; i++)
        cout << ans[i] << " ";
    cout << ans[ans.size()-1] << endl;
    return 0;
}