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

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