AtCoder A - Good Integer の問題で回答の一部にWAが混ざります

Question

前提・実現したいこと

AtCoder Beginner Contest 079の問題で回答の一部にWAが混ざります。
https://atcoder.jp/contests/abc079/tasks/abc079_a

どこを修正すればよいでしょうか？
個人的に試す組み合わせでは全て正答となります。

問題文
1118 のような、3 つ以上の同じ数字が連続して並んだ 4 桁の整数を 良い整数 とします。

4 桁の整数 N が与えられるので、N が 良い整数 かどうかを答えてください。

制約
1000≦N≦9999
入力は整数からなる
入力
入力は以下の形式で標準入力から与えられる。

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

発生している問題・エラーメッセージ

WA

該当のソースコード

#include <bits/stdc++.h>
using namespace std;
 
int main() {
  int N;
  
  cin >> N;
  
  int n1000 = N / 1000;
  int n100 = (N - (n1000 * 1000)) / 100;
  int n10 = (N - ((n1000 * 1000) + (n100 * 100))) / 10;
  int n1 = (N - ((n1000 * 1000) + (n100 * 100) + (n10 * 10))) / 1;
            
  if(n1000 == n100){
  	if(n1000 == n10 || n1000 == n1){
    	cout << "Yes" << endl;
    } 
    else {
    	cout << "No" << endl;
    }
  } 
  else if(n1000 == n10 && n1000 == n1){
  	cout << "Yes" << endl;
  } 
  else if(n100 == n10 && n100 == n1){
  	cout << "Yes" << endl;
  } 
  else {
  	cout << "No" << endl;
  }
            
}

Comments

[Crimson_Tide]

条件は 3 つ以上の同じ数字が”連続して並んだ” 4 桁の整数を 良い整数 をYESにするですね。
始めの判定しか見ていませんが 4桁のうち連続していなくても３つ同じ数字があればYESにしていませんか

[Doggy]

ありがとうございます！
その通りでした。
問題をよく読むことを心がけたいと思います。

[Crimson_Tide]

質問読んでも、こういう質問だろうと思い込んでしまうことどうしてもありますね。

このままだと回答受付中になってしまうので、自己回答して解決済みにしてください。

Answer 1

Crimson_Tideさんのアドバイスに従い、３つ連続していない場合の出力はNoに変更することで解決しました。

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

int main() {
  int N;

  cin >> N;

  int n1000 = N / 1000;
  int n100 = (N - (n1000 * 1000)) / 100;
  int n10 = (N - ((n1000 * 1000) + (n100 * 100))) / 10;
  int n1 = (N - ((n1000 * 1000) + (n100 * 100) + (n10 * 10))) / 1;

  if(n1000 == n100){
      if(n1000 == n10){
        cout << "Yes" << endl;
    } 
    else {
        cout << "No" << endl;
    }
  } 
  else if(n1000 == n10 && n1000 == n1){
      cout << "No" << endl;
  } 
  else if(n100 == n10 && n100 == n1){
      cout << "Yes" << endl;
  } 
  else {
      cout << "No" << endl;
  }

}