USACO - Chapter1 Section 1.4 - Mother's Milk

Mother’s Milk

题目描述

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

写一个程序去帮助农民找出当A桶是空的时候，C桶中牛奶所剩量的所有可能性。

输入输出格式

输入格式：
单独的一行包括三个整数A,B和C。

输出格式：
只有一行，升序地列出当A桶是空的时候，C桶牛奶所剩量的所有可能性。

输入输出样例

输入样例#1：
[输入1]
8 9 10
[输入2]
2 5 10
输出样例#1：
[输出1]
1 2 8 9 10
[输出2]
5 6 7 8 9 10

经典 DFS，把三个桶 A,B,C 的当前牛奶量当一个 cur 状态进行搜索，共有6种可能的再倒的方式。

A -> B, B -> A, …… , C -> A, B -> C

等等，利用一个 vis[mxn][mxn][mxn] 来标记状态的访问，以免进行重复搜索。

Code

#include<iostream>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;
bool vis[25][25][25];
int va,vb,vc;
set<int> s;
void dfs(int a,int b,int c)
{
    if (a == 0 && !s.count(c)) s.insert(c);

    int ab,ac,ba,bc,ca,cb;
    ab = min(a,vb-b);
    ac = min(a,vc-c);
    ba = min(b,va-a);
    bc = min(b,vc-c);
    ca = min(c,va-a);
    cb = min(c,bc-b);
    if (!vis[a-ab][b+ab][c])
    {
        vis[a-ab][b+ab][c] = true;
        dfs(a-ab,b+ab,c);
    }
    if (!vis[a-ac][b][c+ac])
    {
        vis[a-ac][b][c+ac] = true;
        dfs(a-ac,b,c+ac);   
    }
    if (!vis[a+ba][b-ba][c])
    {
        vis[a+ba][b-ba][c] = true;
        dfs(a+ba,b-ba,c);
    }
    if (!vis[a][b-bc][c+bc])
    {
        vis[a][b-bc][c+bc] = true;
        dfs(a,b-bc,c+bc);
    }
    if (!vis[a+ca][b][c-ca])
    {
        vis[a+ca][b][c-ca] = true;
        dfs(a+ca,b,c-ca);
    }
    if (!vis[a][b+cb][c-cb])
    {
        vis[a][b+cb][c-cb] = true;
        dfs(a,b+cb,c-cb);
    }
}
int main()
{
    memset(vis,0,sizeof(vis));
    cin >> va >> vb >> vc;
    dfs(0,0,vc);
    for(set<int>::iterator it = s.begin();it != s.end();it++)
    {
        cout << (*it) << " ";
    }
    return 0;
}