競技プログラミング（AtCoder)

Question

前提・実現したいこと

こちらAtCoderの問題（深さ優先探索問題）
https://atcoder.jp/contests/arc031/tasks/arc031_2
でACのコードに一部コードを追加すると、１つだけWAとなり、その追加したコードによる誤りを教えていただきたい。

追加した理由は前後左右に１つも陸がなければ、探索する必要がなくなると考えたからです。

該当のソースコード

#include "bits/stdc++.h"

using namespace std;
typedef long long ll;
#define rep(i, j, n) for (int i = j; i < n; i++)
#define out(ans) cout << ans << endl;
const long long mod = 1e9 + 7;

int h=10,w=10;
int a[4]={-1,0,1,0},b[4]={0,-1,0,1};
string s[11],c[11];

void dfs(int x,int y){
    rep(i,0,4){
        int tx=x+a[i],ty=y+b[i];
        if(tx>=0&&ty>=0&&tx<w&&ty<h&&c[ty][tx]=='o'){
            c[ty][tx]='x';
            dfs(tx,ty);
        }
    }
}

int main(){

  rep(i,0,h) cin>>s[i];

  rep(i,0,h)rep(j,0,w){
    if(s[i][j]=='o')continue;
//--------------------------ここから
    int num=0;
    rep(i,0,4){
      if(j+a[i]>=0&&i+b[i]>=0&&j+a[i]<10&&i+b[i]<10&&s[i+b[i]][j+a[i]]=='o')num++;
    }
    if(num==0)continue;
//---------------------------ここまでを追加
    rep(k,0,h) c[k]=s[k];
    c[i][j]='o';
    int cnt=0;
    rep(k,0,h)rep(l,0,w){
      if(c[k][l]=='o') {
        c[k][l]='x';
        dfs(l,k);
        cnt++;
      }
    }
    if(cnt==1){
      out("YES");
      return 0;
    }
  }

  out("NO")

  return 0;
}

Answer 1

    rep(i,0,4){
      if(j+a[i]>=0&&i+b[i]>=0&&j+a[i]<10&&i+b[i]<10&&s[i+b[i]][j+a[i]]=='o')num++;
    }

iが、2種類あるのが問題です。

Comments

[houki]

同じ変数に気づきませんでした。ありがとうございます。
元のコードよりも実行時間が短いコードになってACできました！
本当に助かりました^^