蓝桥杯算法提高VIP-质因数2 （C++代码）

#include<iostream>
#include<math.h>
using namespace std;
int sushu(int n){
    int flag=0;
    for(int j=2;j<=sqrt(n);j++){
            if(n%j==0){
                flag=1;
                break;
            }
    }
    if(flag==0){
        return 1;
    }
    return 0;
}
int main(){
    int n,flag,s=2,b,t=0;
    cin>>n;
    b=n;
    while(1){ 
        if(sushu(n)&&n==b){
                cout<<n<<endl;
                t++;
                break;
        }
        else if(sushu(n)){
            cout<<" "<<n<<endl;
            t++;
            break;
        }
        if(n%s==0&&sushu(s)){
            if(n==b){
                cout<<s;
            }
            else{
                cout<<" "<<s;
            }
            n=n/s;
            s=1;
            t++;
        }
        s++;
    }
    cout<<t<<endl;
    return 0;
}