python 三个数判断平方数

原创

mob64ca12e3a791 2024-09-11 04:16:18 ©著作权

文章标签 Python 代码示例 ci 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e3a791的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python中判断三个数是否能形成平方数

在编程中，判断某个条件是否成立是我们常见的需求之一。本文将利用Python语言探讨如何判断三个数字能否组合成一个平方数，并通过代码示例帮助大家理解这一过程。

什么是平方数？

平方数是指一个整数的平方，即可以表示为 ( n^2 ) 的数，其中 ( n ) 是一个整数。比如 1、4、9、16、25……都是平方数。

问题描述

我们针对三个输入的数字（可以为正数、负数或零），判断这三个数的任意两数之和是否为一个平方数。为便于理解，下面是该逻辑的流程：

输入三个数字。
计算这三个数字的任意两两组合的和。
判断每个和是否为平方数。

代码实现

下面是实现这一功能的Python代码示例：

import math

def is_square(n):
    if n < 0:
        return False
    root = math.isqrt(n)
    return root * root == n

def check_square_combination(a, b, c):
    combinations = [
        a + b,
        a + c,
        b + c
    ]

    results = []
    for combo in combinations:
        results.append((combo, is_square(combo)))

    return results

# 示例
a, b, c = map(int, input("请输入三个数字，以空格分隔: ").split())
results = check_square_combination(a, b, c)

# 输出结果
for total, is_square_result in results:
    if is_square_result:
        print(f"{total} 是一个平方数")
    else:
        print(f"{total} 不是一个平方数")

代码解析

is_square(n): 检查一个数是否是平方数。它利用平方根函数 math.isqrt 来计算整型平方根，并验证平方根的平方是否等于原来的数。
check_square_combination(a, b, c): 计算三数中任意两数的和，并返回每个和是否为平方数的结果。具体来说，它将三数的和分别存储在一个列表中，并依次调用 is_square 函数判断是否为平方数。

流程图

为了更好地理解代码的执行流程，我们可以用序列图展示如下：

sequenceDiagram
    participant User
    participant Program

    User->>Program: 输入三个数字
    Program->>Program: 计算任意两数之和
    Program->>Program: 判断每个和是否为平方数
    Program->>User: 输出结果