【算法题】平衡数

牛牛在研究他自己独创的平衡数，平衡数的定义是：将一个数分成左右两部分，分别成为两个新的数。 左右部分必须满足以下两点：
1，左边和右边至少存在一位。 2，左边的数每一位相乘如果等于右边的数每一位相乘，则这个数称为平衡数。
例如：1221这个数，分成12和21的话，1*2=2*1，则称1221为平衡数，再例如：1236这个数，可以分成123和1*2*3=6，所以1236也是平衡数。而1234无论怎样分也不满足平衡数。
输入描述: 输入一个正整数（int范围内）。

输出描述: 如果该数是平衡数，输出 “YES”, 否则输出 “NO”。

输入例子: 1221 1234

输出例子: YES NO

---

先考虑0的个数，如果只有一个0 ，怎么分都不平衡
大于一个0则可以平衡

没有0时，遍历数组判断平衡即可

---

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <utility>

using namespace std;
//#define debug_

void func(string str)
{
    vector<int> vec;
    vec.resize(str.size());
    int left(1), right(1);
    int zeronum(0);

    for (auto i = 0; i < str.size();++i)
    {
        vec[i] = str[i] - '0';
        right *= vec[i];
        if (vec[i] == 0)
        {
            zeronum++;
        }
    }
    if (zeronum==1)
    {
        cout << "NO" << endl;
        return;
    }
    else if (zeronum >1)
    {
        cout << "YES" << endl;
        return;
    }
    right /= vec[0];
    left = left*vec[0];
    for (auto i = 1; i < vec.size();++i)
    {
        left *= vec[i];
        right /= vec[i];
        if (left == right)
        {
            cout << "YES" << endl;
            return;
        }
    }
    cout << "NO" << endl;
}

int main()
{
    string str;
#ifdef debug_
    str = "2121";
    //N = 1234;
#else
    cin >> str;
#endif
    func(str);

    return 0;
}

---