3174 含K个3的数( oj )

描述

输入两个正整数m和k。判断m 能否被19整除，且恰好含有k个3。如果满足条件，则输出YES；否则，输出NO。例如，输入：43833 3，满足条件，输出YES。如果输入：39331 3，尽管有3个3，但不能被19整除，也不满足条件，应输出NO。

输入描述

m 和 k 的值，中间用单个空格间隔。

输出描述

满足条件时输出"YES"，不满足时输出"NO"。

样例输入 1 

43833 3

YES

#include <iostream>
using namespace std;

int countDigit(int num, int digit) {
    int count = 0;
    while (num > 0) {
        if (num % 10 == digit) {
            count++;
        }
        num /= 10;
    }
    return count;
}

int main() {
    int m, k;
    cin >> m >> k;
    if (m % 19 == 0 && countDigit(m, 3) == k) {
        cout << "YES" << endl;
    } else {
        cout << "NO" << endl;
    }
    return 0;
}