C++的一个好玩的小程序

Question

题目描述：
小明发现吃一个苹果可以增加3点幸福度，吃一颗草莓可以增加7点幸福度。小明有一个疑问，他能否通过吃若干个苹果和草莓，使自己的幸福度为N？
输入：
首先是输入数字T，表示接下来有T组数据。（1<=T<=100）
每组数据为一行，包含一个数字N（1<=N<=100）
输出：
如果可以，输出YES，否则输出NO。
样例输入：
2
6
5
样例输出：
YES
NO
提示：6可以由2个3组成，而5不能由3和7组成

Answer 1

#include<iostream>
using namespace std;
// 判断 num = 3 * a + 7 * b
bool isEnable(int num) {
    int maxI = num / 3;
    for (int i = maxI; i >= 0; i--) {
        if ((num - i * 3) % 7 == 0) return true;
    }
    return false;
}
int main() {
    int T; cin >> T;
    int num;
    while (T--) {
        cin >> num;
        cout << (isEnable(num)==1?"YES":"NO") << endl;
    }   
}