본문 바로가기

알고리즘/백준

백준 2251 물통 - 세 쌍 벡터를 이용한 bfs

728x90

https://www.acmicpc.net/problem/2251

 

2251번: 물통

각각 부피가 A, B, C(1≤A, B, C≤200) 리터인 세 개의 물통이 있다. 처음에는 앞의 두 물통은 비어 있고, 세 번째 물통은 가득(C 리터) 차 있다. 이제 어떤 물통에 들어있는 물을 다른 물통으로 쏟아 부을 수 있는데, 이때에는 한 물통이 비거나, 다른 한 물통이 가득 찰 때까지 물을 부을 수 있다. 이 과정에서 손실되는 물은 없다고 가정한다. 이와 같은 과정을 거치다보면 세 번째 물통(용량이 C인)에 담겨있는 물의 양이 변할 수도 있다.

www.acmicpc.net

이 문제를 풀기 위해서는 물통 a,b,c를 각각 한 쌍으로 만든 다음에, 세쌍 동시에 bfs를 진행해야 한다. 

a b c 각각 물통에 담긴 물의 양을 저장해가면서 bfs를 진행할 것.

 

 

 

#include <vector>
#include <algorithm>
#include <queue>
#include <iostream>
using namespace std;
int A,B,C;
bool visited[201][201][201];

vector bfs(void){
    queue<pair<pair<int,int>,int>> q;
    q.push(make_pair(make_pair(0,0),C));
    vector result;

    while(!q.empty()){
        int a = q.front().first.first;
        int b = q.front().first.second;
        int c = q.front().second;
        q.pop();

        if(visited[a][b][c]){
            continue;
        }
        visited[a][b][c]=true;
        if(a==0)
            result.push_back(c);

        //a->b
        if (a + b > B) //안넘치는 범위에서 따라야함
            q.push(make_pair(make_pair(a + b - B, B), c));
        else //걍따라도 됨
            q.push(make_pair(make_pair(0, a + b), c));



        //a->c
        if (a + c > C)
            q.push(make_pair(make_pair(a + c - C, b), C));
        else
            q.push(make_pair(make_pair(0, b), a + c));



        //b->a
        if (b + a > A)
            q.push(make_pair(make_pair(A, b + a - A), c));
        else
            q.push(make_pair(make_pair(b + a, 0), c));



        //b->c
        if (b + c > C)
            q.push(make_pair(make_pair(a, b + c - C), C));
        else
            q.push(make_pair(make_pair(a, 0), b + c));



        //c->a
        if (c + a > A)
            q.push(make_pair(make_pair(A, b), c + a - A));
        else
            q.push(make_pair(make_pair(c + a, b), 0));



        //c->b
        if (c + b > B)
            q.push(make_pair(make_pair(a, B), c + b - B));
        else
            q.push(make_pair(make_pair(a, c + b), 0));
    }
    return result;
}


int main(void){
    cin>>A>>B>>C;
    vector result = bfs();
    sort(result.begin(),result.end());
    for (int i = 0; i < result.size(); i++)
        cout << result[i]<< " ";

}