C语言判断5位数是否为素数

#include <stdio.h>

int main() {
    int num, i, flag = 1;
    printf("请输入一个5位数：");
    scanf("%d", &num);
    for(i = 2; i < num; i++) {
        if(num % i == 0) {
            flag = 0; // 不是素数
            break;
        }
    }
    if(num == 1 || num == 0) {
        flag = 0;
    }
    if(flag == 1) {
        printf("%d是素数", num);
    } else {
        printf("%d不是素数", num);
    }
    return 0;
}