AtCoder Beginner Contest 350 A問題

AtCoder Beginner Contest 350 A問題
こちらの問題です。

問題

入出力例

解法

例外である316と0を除き、最後の3文字が349以下であるかどうか確認します。

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

int main() {
    string input;
    cin >> input;
    int len = input.length();
    
    string lastThreeDigits = input.substr(len - 3);
    int number = stoi(lastThreeDigits);

    if ((number != 0 && number != 316) && (number <= 349)) {
        cout << "Yes" << endl;
    } else {
        cout << "No" << endl;
    }

    return 0;
}