POJ 3278 Catch That Cow(BFS廣度優先搜尋)

//BFS廣度優先搜尋，結合資料結構隊列<br />#include<iostream><br />#include<queue><br />using namespace std;<br />int arr[200000];<br />int vis[200000];//已訪問數組<br />queue<int> q1,q2;//用資料結構隊列可節省搜尋空間<br />int deep;<br />void set(int n,int deep,int k)<br />{<br />if(arr[n-1] != deep-1 && n - 1 > 0 && !vis[n-1])//控制搜尋範圍，剪枝<br />{<br />arr[n-1] = deep;<br />q2.push(n-1);<br />}<br />if(arr[n+1] != deep-1 && !vis[n+1])<br />{<br />arr[n+1] = deep;<br />q2.push(n+1);<br />}<br />if(arr[2*n] != deep-1 && n < k && !vis[2*n])//控制搜尋範圍<br />{<br />arr[2*n] = deep;<br />q2.push(2*n);<br />}<br />}<br />int bfs(int n,int k)//廣度優先搜尋——針對樹的最淺深度<br />{<br />arr[n] = 1;<br />q1.push(n);<br />if(n >= k)//當農夫位置在牛的位置的後面時，直接相減得出結果<br />return n - k;<br />else<br />{<br />while(arr[k] == 0)<br />{<br />while(!q1.empty())<br />{<br />if(!vis[q1.front()])<br />{<br />set(q1.front(),deep,k);<br />vis[q1.front()] = 1;<br />q1.pop();<br />}<br />elseq1.pop();<br />}<br />while(!q2.empty())//將隊列複製過去<br />{<br />q1.push(q2.front());<br />q2.pop();<br />}<br />++deep;<br />}<br />return arr[k]-1;<br />}<br />}<br />int main()<br />{<br />int n,k;<br />cin >> n >> k;<br />deep = 2;<br />memset(arr,0,sizeof(arr));<br />memset(vis,0,sizeof(vis));<br />cout << bfs(n,k) << endl;<br />return 0;<br />}