NYOJ21.三个水杯-初始态到目标态的最少次数-经典BFS

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<queue>
 4 #include<cstring>
 5 #include<algorithm>
 6 using namespace std;
 7 const int maxn=100+10;
 8 int vis[maxn][maxn][maxn];
 9 typedef struct node{
10     int cur[3];
11     int step;
12 };
13 node s,e,p;
14 int v[3];
15 bool match(node t){
16     return (t.cur[0]==e.cur[0]&&t.cur[1]==e.cur[1]&&t.cur[2]==e.cur[2]);
17 }
18 int BFS(){
19     s.cur[0]=v[0];
20     s.cur[1]=s.cur[2]=s.step=0;
21     queue<node>q;
22     memset(vis,0,sizeof(vis));
23     q.push(s);
24     vis[v[1]][0][0]=1;
25     while(!q.empty()){
26       p=q.front();q.pop();
27       if(match(p))return p.step;
28       for(int i=0;i<3;i++){
29         for(int j=0;j<3;++j){
30             if(i!=j){
31                 int minn=min(v[j]-p.cur[j],p.cur[i]);
32                 node v=p;
33                 v.cur[j]+=minn;
34                 v.cur[i]-=minn;
35                 v.step=p.step+1;
36                 if(vis[v.cur[0]][v.cur[1]][v.cur[2]]==0){
37                     q.push(v);
38                     vis[v.cur[0]][v.cur[1]][v.cur[2]]=1;
39                 }
40             }
41         }
42       }
43     }
44     return -1;
45 }
46 int main(){
47     int t;
48     scanf("%d",&t);
49     while(t--){
50         scanf("%d%d%d",&v[0],&v[1],&v[2]);
51         scanf("%d%d%d",&e.cur[0],&e.cur[1],&e.cur[2]);
52         int ans=BFS();
53         printf("%d\n",ans);
54     }
55     return 0;
56 }