平衡数

平衡数的定义：将一个数分成左右两部分，分别成为2个新数。左右不分必须满足：

1、左边和右边至少存在一位

2、左边数的每一位相乘如果等于右边数每一位相乘

则这个数称为平衡数。

如：1221这个数，分成12和21，1*2 = 2*1，是个平衡数。

1236这个数，分成123和6，1*2*3 = 6，是个平衡数。

输入：一个正整数（int范围）

输出：如果是平衡数输出“YES”，否则输出“NO”。

#include <iostream>
#include <vector>
using namespace std;

bool isBalance(int n){
    if(n < 10){
        return false;
    }
    int count = 0;
    int temp = n;   //计算位数
    while(temp != 0){
        temp/=10;
        count++;
    }
    vector<int> ret;   //由低位到高位放入容器内
    while(n != 0){
        ret.push_back(n%10);
        n/=10;
    }
    
    int flag = false;
    int mult1 = 1;   
    for(int i=0; i<count-1; i++){   //循环相乘判断是否相等
        mult1 *= ret[i];   //右边相乘结果
        int mult2 = 1;   
        for(int j=i+1; j<count; j++){
            mult2 *= ret[j];  //左边相乘结果
        }
        if(mult1 == mult2){
            flag = true;
            break;
        }
    }
    if(flag){
        return true;
    }
    else{
        return false;
    }
}

int main(){
    int input;
    cin >> input;
    
    if(isBalance(input)){
        cout << "YES";
    }
    else{
        cout << "NO";
    }
    return 0;
}