平衡数

平衡数的定义：将一个数分成左右两部分，分别成为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;
}