A section of school recruitment and test questions 1-a-share online innovation

A section of the school entrance examination questions 1-A Play new on the stock line

【 Input 】
Enter the assignment start number in the first line num, Number of allotments x.
In the next few lines, enter n Hemo n digit , Space between .
Input END end .
Such as ：
12345 5
2 12 23
3 123 345
END
【 Output 】
Judge whether the lot is won . At the end of the n The number of digits and the end of the new number n If the digits match, the lot will be won , Output YES, Then output the winning numbers in turn . Otherwise output NO.
Such as ：
YES
345
【 notes 】
Call a new number {num,num+1,…,num+x-1}.

#include <bits/stdc++.h>
using namespace std;

//  Split string 
vector<string> splitSpace(string& str, char ch)
{
    
    vector<string> ans;
    string ansstr;
    for (int i = 0; i < str.size(); i++)
    {
    
        if (str[i] == ch)
        {
    
            if (!ansstr.empty())
            {
    
                ans.push_back(ansstr);
                ansstr.clear();
            }
            continue;
        }
        if (i == str.size() - 1 && isdigit(str[i]))
        {
    
            ansstr += str[i];
            ans.push_back(ansstr);
            break;
        }
        ansstr += str[i];
    }
    return ans;
}

//  String to integer 
int str2int(string& str)
{
    
    int ans = 0, slope = 1;
    for (int i = str.size() - 1; i >= 0; i--)
    {
    
        ans += (str[i] - '0') * slope;
        slope *= 10;
    }
    return ans;
}

//  String addition 
string stradd(string& str1, string& str2)
{
    
    string ans;
    int cur = 0, i = str1.size() - 1, j = str2.size() - 1;
    while (i >= 0 || j >= 0 || cur != 0)
    {
    
        if (i >= 0)
            cur += str1[i--] - '0';
        if (j >= 0)
            cur += str2[j--] - '0';
        if (cur % 10 >= 0)
            ans += to_string(cur % 10);
        cur /= 10;
    }
    reverse(ans.begin(), ans.end());
    return ans;
}

int main()
{
    
    string input;
    getline(cin, input);
    vector<string> numX = splitSpace(input, ' ');
    string num = numX[0];
    int x = str2int(numX[1]);
    vector<string> ans;
    do
    {
    
        input.clear();
        getline(cin, input);
        if (input == string("END"))
        {
    
            break;
        }
        vector<string> array = splitSpace(input, ' ');
        int tail = str2int(array[0]);
        for (int i = 1; i < array.size(); i++)
        {
    
            string match = num;
            for (int k = 0; k < x; k++)
            {
    
                string sk = to_string(k);
                match = stradd(num, sk);
                string subs = match.substr(match.size() - tail, match.size() - 1);
                if (subs == array[i])
                {
    
                    ans.push_back(array[i]);
                }
            }
        }
    } while (true);
    if (ans.empty())
    {
    
        cout << "NO" << endl;
    }
    else
    {
    
        cout << "YES" << endl;
        for (auto &&e : ans)
        {
    
            cout << e << " ";
        }
        cout << endl;
    }
    return 0;
}