判断平方增一数 - Python 代码实现 - 常规

判断平方增一数 - Python 代码实现

定义: 若一个正整数有 2n 个数位，后 n 位组成的数恰好比前 n 位组成的数大 1，则这个数称为增一数。例如 34、2526、233234 都是增一数。如果这个数还是某个数的平方，则称为平方增一数。

任务: 你的任务就是判断任一给定正整数是否平方增一数。

输入格式: 输入在第一行中给出一个正整数 N (≤100)，随后 N 行，每行给出一个不超 2^31 的待判定的正整数。

输出格式: 对每个待判定的正整数，在一行中输出判定结果：

如果是平方增一数，则输出 2；
如果只是普通增一数，则输出 1；
如果不是增一数，则输出 0。

样例输入:

样例输出:

2
1
0

Python 代码:

n = int(input())
for i in range(n):
    num = input().strip()
    l = len(num)
    if l == 1:
        if num == '0' or num == '1' or num == '2':
            print(1)
        else:
            print(0)
    elif l % 2 == 0:
        a = int(num[:l//2])
        b = int(num[l//2:])
        if b - a == 1:
            if int(num) == (a+b)**2:
                print(2)
            else:
                print(1)
        else:
            print(0)
    else:
        a = int(num[:l//2])
        b = int(num[l//2:])
        if b - a == 1:
            if int(num) == (a+1)*10**(l//2) + a + 1:
                print(2)
            else:
                print(1)
        else:
            print(0)

代码解释:

首先读取输入的整数 N 和 N 个待判定的正整数。
对于每个正整数，判断其长度是否为奇数或偶数。
如果长度为奇数，则将该数分为前后两部分，分别为前 (l//2) 位和后 (l//2) 位，判断后半部分是否比前半部分大 1。如果是，则再判断该数是否为某个数的平方，如果是则输出 2，否则输出 1；否则输出 0。
如果长度为偶数，则将该数分为前后两部分，分别为前 (l//2) 位和后 (l//2) 位，判断后半部分是否比前半部分大 1。如果是，则再判断该数是否为某个数的平方，如果是则输出 2，否则输出 1；否则输出 0。
如果长度为 1，则判断该数是否为 0、1 或 2，如果是则输出 1，否则输出 0。

代码示例:

输出:

2
1
0

代码说明:

代码使用了 Python 的字符串切片功能，将输入的正整数分成前后两部分。
代码使用了 Python 的整数类型转换功能，将字符串类型的数字转换成整数类型。
代码使用了 Python 的算术运算符，计算数字的平方。
代码使用了 Python 的条件语句，判断数字是否满足增一数和平方增一数的定义。

代码特点:

代码简洁易懂，易于理解和维护。
代码效率较高，能够快速地判断数字是否为平方增一数。
代码可扩展性强，可以方便地修改代码以满足不同的输入格式和输出格式要求。