7-8 正整数A+B

7-8 正整数A+B

作者 陈越

单位 浙江大学

题的目标很简单，就是求两个正整数A和B的和，其中A和B都在区间[1,1000]。稍微有点麻烦的是，输入并不保证是两个正整数。

输入格式：

输入在一行给出A和B，其间以空格分开。问题是A和B不一定是满足要求的正整数，有时候可能是超出范围的数字、负数、带小数点的实数、甚至是一堆乱码。

注意：我们把输入中出现的第1个空格认为是A和B的分隔。题目保证至少存在一个空格，并且B不是一个空字符串。

输出格式：

如果输入的确是两个正整数，则按格式A + B = 和输出。如果某个输入不合要求，则在相应位置输出?，显然此时和也是?。

输入样例1：

123 456

输出样例1：

123 + 456 = 579

输入样例2：

22. 18

输出样例2：

? + 18 = ?

输入样例3：

-100 blabla bla...33

输出样例3：

? + ? = ?

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

代码

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

int main()
{
    string inputNum;
    getline(cin, inputNum);
    int namespace1 = -1;
    bool ifValidA = true, ifValidB = true;
    int aNum = 0, bNum = 0;
    // 寻找第一个空格
    for (int i = 0; i < inputNum.size(); i++)
    {
        if (inputNum[i] == ' ')
        {
            namespace1 = i;
            break;
        }
    }
    for (int i = 0; i < namespace1; i++)
    {
        if (inputNum[i] < '0' || inputNum[i] > '9')
        {
            ifValidA = false;
            break;
        }
        else
        {
            aNum = aNum * 10 + (inputNum[i] - '0');
        }
    }
    for (int i = namespace1 + 1; i < inputNum.size(); i++)
    {
        if (inputNum[i] < '0' || inputNum[i] > '9')
        {
            ifValidB = false;
            break;
        }
        else
        {
            bNum = bNum * 10 + (inputNum[i] - '0');
        }
    }
    // 判断两个数字是否在规定的范围[1,1000]内
    if (aNum < 1 || aNum > 1000)
        ifValidA = false;
    if (bNum < 1 || bNum > 1000)
        ifValidB = false;
    // 开始输出
    if (ifValidA)
    {
        cout << aNum;
    }
    else
    {
        cout << "?";
    }
    cout << " + ";
    if (ifValidB)
    {
        cout << bNum;
    }
    else
    {
        cout << "?";
    }
    if (!ifValidA || !ifValidB)
    {
        cout << " = ?";
    }
    else
        cout << " = " << aNum + bNum;
    return 0;
}

优化代码

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

int main()
{
    string inputNum;
    getline(cin, inputNum);
    // 寻找第一个空格
    size_t spacePos = inputNum.find(' ');
    if (spacePos == string::npos) // 如果不存在空格
    {
        cout << "? + ? = ?" << endl;
        return 0;
    }

    int aNum = 0, bNum = 0;
    bool ifValidA = true, ifValidB = true;
    // 第一个数字
    for (size_t i = 0; i < spacePos; i++)
    {
        if (!isdigit(inputNum[i]))
        {
            ifValidA = false;
            break;
        }
        aNum = aNum * 10 + (inputNum[i] - '0');
    }
    if (!ifValidA || aNum < 1 || aNum > 1000)
        ifValidA = false;
    // 第二个数字
    for (size_t i = spacePos + 1; i < inputNum.length(); i++)
    {
        if (!isdigit(inputNum[i]))
        {
            ifValidB = false;
            break;
        }
        bNum = bNum * 10 + (inputNum[i] - '0');
    }
    if (!ifValidB || bNum < 1 || bNum > 1000)
        ifValidB = false;
    cout << (ifValidA ? to_string(aNum) : "?") << " + "
         << (ifValidB ? to_string(bNum) : "?") << " = "
         << ((ifValidA && ifValidB) ? to_string(aNum + bNum) : "?") << endl;
    return 0;
}

• 使用size_t来表示数组的大小或数组的索引
• 使用string的find()函数来寻找第一个空格的位置
• 使用三元表达式而不是if-else语句来输出，简化了代码，更加高效
• 使用isdigit()来更高效的判断是否为数字