华为OD 2024笔试机试 - 最多几个直角三角形 (python C卷D卷真题算法)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

有N条线段，长度分别为a[1]-a[n]。

现要求你计算这N条线段最多可以组合成几个直角三角形。

每条线段只能使用一次，每个三角形包含三条线段。

输入描述

第一行输入一个正整数T（1<＝T<＝100），表示有T组测试数据.

对于每组测试数据，接下来有T行，

每行第一个正整数N，表示线段个数（3<＝N<＝20），接着是N个正整数，表示每条线段长度，（0<a[i]<100）。

输出描述

对于每组测试数据输出一行，每行包括一个整数，表示最多能组合的直角三角形个数

用例

输入

1
7 3 4 5 6 5 12 13

输出

2

说明 可以组成2个直角三角形（3，4，5）、（5，12，13）

解题思路

1. 排序：将每组测试数据中的线段按长度排序。
2. 枚举：使用三重循环枚举所有可能的三条线段组合 (a, b, c)，其中 a <= b <= c。
3. 判断直角三角形：检查这些组合是否满足直角三角形的条件，即 a^2 + b^2 == c^2。
4. 记录已使用的线段：使用一个布尔数组记录哪些线段已经被使用，确保每条线段只能使用一次。
5. 计数：每找到一个符合条件的直角三角形，就计数加一。

参考代码

def is_right_triangle(a, b, c):
    return a * a + b * b == c * c

def max_right_triangles(segments):
    segments.sort()
    used = [False] * len(segments)
    count = 0

    for i in range(len(segments) - 1, 1, -1):
        if used[i]:
            continue
        for j in range(i - 1, 0, -1):
            if used[j]:
                continue
            for k in range(j - 1, -1, -1):
                if used[k]:
                    continue
                if is_right_triangle(segments[k], segments[j], segments[i]):
                    used[k] = used[j] = used[i] = True
                    count += 1
                    break
    return count

def main():
    import sys
    input = sys.stdin.read
    data = input().split()
    
    T = int(data[0])
    index = 1
    results = []
    
    for _ in range(T):
        N = int(data[index])
        segments = list(map(int, data[index + 1: index + 1 + N]))
        index += N + 1
        results.append(max_right_triangles(segments))
    
    for result in results:
        print(result)

if __name__ == "__main__":
    main()