之前做过的一道dfs简单题,但是这次交还是WA了几发RE了几发,然后总结了下dfs应该注意的问题。
1.注意dfs的结束条件。
2.要有标记量(vis数组),防止无限递归(RE所在)。
<注意多组输入orz>
上代码:
#include <bits/stdc++.h>
#define me(x,y) memset(x,y,sizeof(x))
#define sd(x) scanf("%d",&x)
#define ss(x) scanf("%s",x)
#define sf(x) scanf("%f",&x)
#define slf(x) scanf("%lf",&x)
#define pd(x) printf("%d\n",x)
#define plld(x) printf("%lld\n",x)
#define ps(x) printf("%s\n",x)
#define max(x,y) (x>=y?x:y)
#define min(x,y) (x<y?x:y)
#define sum(x,y) (x+y)
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
string s;
int m[30][30];
//dfs记得设置标记量防止无限递归
bool vis[30];
int flag;
void dfs(int j) {
if(flag)return ;
if(j=='m'-'a') {
flag=1;
return ;
}
vis[j]=true;
for(int i=0;i<26;i++) {
if(flag)return ;
if(m[j][i] && !vis[i]) dfs(i);
}
}
int main () {
while(1) {
me(m,0);
me(vis,false);
while(cin>>s&&s[0]!='0') m[s[0]-'a'][s[s.length()-1]-'a']=1;
if(cin.eof())break;
flag=0;
dfs('b'-'a');
if(!flag)cout<<"No."<<endl;
else cout<<"Yes."<<endl;
}
return 0;
}