华为OD机考直角三角形数量 搞出来了

Java直角三角形问题

 

java实现有N条线段，长度分别为a[1]-a[n]。 现要求你计算这N条线段最多可以组合成几个直角三角形。 每条线段只能使用一次，每个三角形包含三条线段。 输入描述 第一行输入一个正整数T（1<＝T<＝100），表示有T组测试数据. 对于每组测试数据，接下来有T行， 每行第一个正整数N，表示线段个数（3<＝N<＝20），接着是N个正整数，表示每条线段长度，（0<a[i]<100）。 输出描述 对于每组测试数据输出一行，每行包括一个整数，表示最多能组合的直角三角形个数 示例1 输入输出示例仅供调试，后台判题数据一般不包含示例 输入 1 7 3 4 5 6 5 12 13 输出 2

import java.util.*;

public class Main {
    // 线段结构体
    static class Segment {
        int length; // 长度
        boolean used; // 是否被使用

        public Segment(int length) {
            this.length = length;
            this.used = false;
        }
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int t = scanner.nextInt(); // 测试数据组数
        while (t-- > 0) {
            int n = scanner.nextInt(); // 线段数量
            Segment[] segments = new Segment[n]; // 存储线段的数组
            for (int i = 0; i < n; i++) {
                segments[i] = new Segment(scanner.nextInt());
            }
            Arrays.sort(segments, (a, b) -> a.length - b.length); // 按照长度升序排序
            int ans = 0; // 最多能组合的直角三角形个数
            for (int i = n - 1; i >= 0; i--) {
                segments[i].used = true; // 标记当前线段为已使用
                for (int j = i - 1; j >= 0; j--) {
                    if (!segments[j].used) {
                        for (int k = j - 1; k >= 0; k--) {
                            if (!segments[k].used) {
                                int a = segments[i].length;
                                int b = segments[j].length;
                                int c = segments[k].length;
                                if (a * a == b * b + c * c) { // 满足直角三角形条件
                                    ans++;
                                    segments[j].used = true; // 标记已使用的线段
                                    segments[k].used = true;
                                    break;
                                }
                            }
                        }
                    }
                }
            }
            System.out.println(ans);
        }
    }
}