Подсчитать количество цифр k в заданном натуральном числе, где k – случайная цифра

#include "stdafx.h"
#include <iostream>
#include <time.h>
#include <string>
 
using namespace std;
 
int main()
{
    srand(time(NULL));
    setlocale(LC_ALL, "");
    int n;
    cout << "Введите n: ";
    cin >> n;
    string strN = to_string(n); //преобразовываем n в строку
 
    short k = rand() % 10; // генерируем рандомное число k, от 0 до 9, т.к. это должна быть цифра
    string strK = to_string(k); // преобразовываем в строку
 
    short p = 0, prov = 0;
    while (true)
    {
        prov = strN.find(strK, prov + 1); // ищем k в n, начиная с позиции prov + 1
        if (prov != -1) p++; // если нашли - увеличиваем счетчик 
        else break;
    }
    cout << "В числе " << n << " " << p << " цифр " << k << endl;
    system("pause");
}

#include <algorithm>
#include <iostream>
#include <random>
#include <string>
 
int main()
{
    const int N = 1234321;
    const std::string str = std::to_string(N);
    std::mt19937 gen{ std::random_device()() };
    const int k = std::uniform_int_distribution<>(0, 9)(gen);
    std::cout << k << ": " << std::count(str.begin(), str.end(), k + '0') << std::endl;
}