2018.8.5拼多多笔试题2———数字分组

输入描述：输入为一行，包含一个字符串形式的正整数
输出描述：输出为一行，找出经过“拆分”和“变化”后的所有组合数目
拆分操作：将一个字符串拆分成两部分
变化操作：分别在这两部分中加上一个小数点或者不加；
最后分别形成一个整数和小数
形成的小数和整数满足以下要求：
1. 不可包含多余的前缀0，即010和010.1不合法
2. 小数，不可包含多余的后缀0，如0.10不合法
3. .1和1.也不合法

输入示例：123
输出示例：4
说明：【1，23】，【1，2.3】，【12，3】,【1.2，3】

输入示例：00011
输出示例：2
说明：【0.001，1】，【0，0.011】

#include <iostream>
#include <string>
#include <algorithm>
#include <stdio.h>
#include <vector>
using namespace std;

int strSplit(string s)
{
    if(s.length()==1)
        return 1;
    if(s[0] == '0' && s[s.length()-1] == '0') 
        return 0;
    if(s[0] == '0' || s[s.length()-1] == '0') 
        return 1;
    return s.length();
}

int DigitGroup(string str) {
    int res = 0;
    for(int i=1;i<str.length();i++)
    {
        int a=strSplit(str.substr(0,i));
        int b=strSplit(str.substr(i));
        res+=a*b;
    }
    return res;
}

int main()
{
    string str;
    cin>>str;
    int count = DigitGroup(str);
    cout<<count<<endl;
    return 0;
}