POJ 3414 Pots【BFS＋路径】

链接: POJ3414

题意

给定两个a,b两个容器的体积,以及目标体积c,通过以下的操作:

1. FILL（i） 装满容器 i
2. DROP（i） 将容器 i 倒掉
3. POUR（i，j）从容器 i 倒到容器 j

要求能在有限的步骤下达到目标体积c,输出其最短的步骤序列,否则输出:“impossible”.

思路

求最短序列,很明显这是个状态空间问题,用BFS求解即可.
根据题意我们知道有6个状态:
体积: a, b
由当前状态(x, y)得到:

1. FILL(x) 将x倒满 (a, y)
2. FILL(y) 将y倒满 (x, b)
3. DROP(x) 倒掉x (0, y)
4. DROP(y) 倒掉b (x, 0)
5. POUR(x,y) 从y倒到x (x+min(a-x,y), y-min(a-x,y))
6. POUR(y,x) 从x倒到y (x-min(b-y,x), y+min(b-y,x))

用vis记录状态,因为需要输出序列的每个步骤,需要保存路径 root当前节点->根节点.

AC代码

#include <cstdio>
#include <iostream>
#include <queue>

using namespace std;
typedef pair<int,int> P;

P root[101][101];//保存路径
int vis[101][101], op[101][101], sign[101][101] ;
//vis标记 op存0~2执行的操作  sign 容器i
int a,b,c;


void judge(int flag, int x){
    if(flag == 0) printf("FILL(%d)\n",x);
    else if(flag == 1) printf("DROP(%d)\n",x);
    else printf("POUR(%d,%d)\n",x/10,x%10);    
}

void link(P p){
    if(p == P()) return;;
    link(root[p.first][p.second]);
    judge(op[p.first][p.second], sign[p.first][p.second]);
}

void BFS(){
    queue<P> que;
    que.push(P(0,0));
    vis[0][0] = 1;
    while(!que.empty()){
        P s = que.front(); que.pop();
        int x= s.first, y = s.second;
        if(x == c || y == c){
            printf("%d\n",vis[x][y]-1);
            link(P(x,y));//路径
            return;
        }
        int k = min(a-x,y), k1 = min(b-y,x);
        int st[6][2] = {{a,y},{x,b},{0,y},{x,0},{x+k,y-k},{x-k1,y+k1}};//状态
        int oper[6] ={1,2,1,2,21,12};//容器i
        for(int i = 0; i < 6;i++){
            int dx = st[i][0], dy = st[i][1];
            if(!vis[dx][dy]){
                vis[dx][dy] = vis[x][y]+1;
                root[dx][dy] = P(x,y); 
                sign[dx][dy] = oper[i];
                op[dx][dy] = i/2;
                que.push(P(dx,dy));
            }
        }
    }
    printf("impossible\n");
}

int main(){
    scanf("%d%d%d",&a, &b, &c);
    BFS();
    return 0;
}