AtCoder A - Good IntegerでWAになる原因がわかりません

Question

AtCoder Beginner Contest 079の A-Good IntegerがWAになって困っています。
https://atcoder.jp/contests/abc079/tasks/abc079_a
14個のテストケースのうち1ケースだけWAになってしまうのですが、テストケースの中身が公開されていないためコードの修正箇所がわかりません。

問題文
1118 のような、3 つ以上の同じ数字が連続して並んだ 4 桁の整数を 良い整数 とします。
4 桁の整数 N が与えられるので、N が 良い整数 かどうかを答えてください。

制約
1000≦N≦9999
入力は整数からなる

出力
N が 良い整数 ならば Yes を、そうでなければ No を出力せよ。

#include <bits/stdc++.h>
using namespace std;

int main() {
  int n;
  cin>>n;
  int c=1000;
  int b=0;
  int ans=0;
  for(int i=0;i<4;i++){
    if(b==n/c){
      ans++;
    }
    b=n/c;
    n-=n/c*c;
    c/=10;
  }
  if(ans>1){
    cout<<"Yes"<<endl;
  }
  else{
    cout<<"No"<<endl;
  }
}

Answer 1

たとえば1122など2連続でもYesと誤答します。
提示コードでの判定の仕組みがよく理解できていませんが
同じ数が連続する回数を数えるのはややコードが煩雑になりそうなので
各桁の出現頻度を0～9までの10個の配列に記録して、最後に最高の頻度が3かを判定したほうが簡便かと思います。
単に出現頻度を求めても意味ないですね。

連続回数を数え上げる方法でもコードはシンプルに書けそうです。
以下C++でもほぼ同じく書けます。

def is_good(n):
    cnt, pre = 0, -1
    while n:
        if cnt >= 3:
            return True
        cur = n % 10
        cnt = cnt+1 if pre == cur else 1 # 連続していたら+1、途切れたら1
        pre = cur
        n //= 10
    return cnt >= 3

for n in range(1111,10000):
    print( n, is_good(n))

Comments

[ppp_ha]

ありがとうございます！

Answer 2

1122などがYesに誤判定されます

Comments

[ozwk]

大きい桁から見ていって、前の桁と同じであればans++;をして、ansが2以上かを見ているようですが、
このやり方だと2連続x2もYesと判定してしまいます。

Answer 3

質問のコードの問題点は既に指摘されていますが、
入力の 4桁の整数を文字列で読み込めば、
3 つ以上の同じ数字が連続して並んでいるかどうかは、
s[0]=s[1]=s[2] または s[1]=s[2]=s[3] で判定できますよね。

#include <iostream>
#include <string>
using namespace std;

int main()
{
	string s;
	cin >> s;
	cout << (s[1]==s[2] && (s[0]==s[1] || s[2]==s[3]) ? "Yes\n" : "No\n");
}