poj 2362 dfs
阿新 • • 發佈:2018-12-05
題意:
給你M根長度可能不同的棍子,問你用著M根棍子能否拼成一個正方形.
#include<iostream> #include<cmath> #include<cstring> #include<algorithm> using namespace std; int sum; int m; int a[30]; int vis[30]; int dfs(int cnt,int left,int be) //cnt當前是第幾條邊,left是這條邊還需要多長,be是從第幾條邊開始搜。 { if(cnt==4)return 1; //只要有三條邊找到了,就完成了。 else for(int i=be;i<=m;i++)if(vis[i]==0) { if(a[i]==left) { vis[i]=1; if(dfs(cnt+1,sum/4,1))return 1; vis[i]=0; } if(a[i]<left) { vis[i]=1; if(dfs(cnt,left-a[i],i+1))return 1;//這裡為什麼從第i+1開始。請看下面分析 vis[i]=0; } } return 0; } int main() { int t; cin>>t; while(t--) { int max_len=0; sum=0; memset(vis,0,sizeof(vis)); cin>>m; for(int i=1;i<=m;i++) { cin>>a[i]; sum+=a[i]; max_len=max(max_len,a[i]); } if(sum%4!=0||max_len>sum/4) { cout<<"no"<<endl; } else { if(dfs(1,sum/4,1)) { cout<<"yes"<<endl; } else { cout<<"no"<<endl; } } } return 0; }
假設我們當前剛設定完vis[5]=1,表示第5根棍子我們在還是left長度的時候用了,現在我們還是left-len[5]長度的時候要不要去用len[2]來嘗試配對一下?不用,肯定無解.因為搜尋的時候我們是先找的len[2]的,如果len[2]+len[5]+….有一個可行解(可以配對成當前cnt邊)的話,我們之前肯定已經找到了且推出了dfs這個函式.但是現在明顯我們還沒找到,所以len[2]+len[5]+…不可能是一個可行解,所以直接從5後面的位置繼續找即可.如果還不能理解,就把本題想象成我們只需要配對一條正方形的邊,不用配4邊了.