挤成一坨的代码

#include <cstdio>
#include <cmath>
bool prime(int n){if(n==1)return 0;else if(n==2||n==3)return 1;else if(n%6 != 5 && n%6 != 1)return 0;for(int i=5;i<sqrt(n);i+=6){if(n%i == 0 || n%(i+2) == 0)return 0;}return 1;}
int main(){long n;scanf("%d",&n);prime(n)?printf("Yes"):printf("No");}