Проверка в числе на три повторяющиеся цифры

/////////////////////////////////////////////////////////////////////
// Дано четырехзначное целое число. 
// Определить содержит, ли число ровно три одинаковые цифры.
 
// Пример входных данных
// 1234
// 1114
// 1211
// Пример выходных данных
// NO
// YES
// YES
/////////////////////////////////////////////////////////////////////
#include <iostream>
#include <algorithm>
/////////////////////////////////////////////////////////////////////
int main()
{
    int n{};
    
    std::cin >> n;
    
    int nd[4]{ n / 1000, n / 100 % 10, n / 10 % 10, n % 10  };
    
    if  ( 
            std::count( nd, nd + 4, nd[0] ) == 3 
         || std::count( nd, nd + 4, nd[1] ) == 3 
        )
    {
        std::cout << "YES";
    }
    else
    {
        std::cout << "NO";
    }
}

#include <iostream>
 
int main()
{
int index[10] = {0};
long long int number;
std::cout << "Vvedite thislo: ";
std::cin >> number;
while (number > 0)
{
index[number%10]++;
number/=10;
}
bool  flag = false;
for (int i = 0; i<10; i++) 
if (index[i] == 3 )  flag = true;
std::cout << (flag? "Yes" : "No");
 
std::cin.get();
return 0;       
}