蓝桥杯算法提高VIP-质因数 （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;
    cin>>n;
    b=n;
    while(1){ 
        if(sushu(n)&&n==b){
                cout<<n<<"="<<n<<endl;
                break;
        }
        else if(sushu(n)){
            cout<<"*"<<n<<endl;
            break;
        }
        if(n%s==0&&sushu(s)){
            if(n==b){
                cout<<b<<"="<<s;
            }
            else{
                cout<<"*"<<s;
            }
            n=n/s;
            s=1;
        }
        s++;
    }
    
    return 0;
}