P5728 【深基5.例5】旗鼓相当的对手---java

题目描述

现有 N(N≤1000)N(N\le 1000)N(N≤1000) 名同学参加了期末考试，并且获得了每名同学的信息：语文、数学、英语成绩（均为不超过 150 的自然数）。如果某对学生 <i,j> 的每一科成绩的分差都不大于 5，且总分分差不大于 10，那么这对学生就是“旗鼓相当的对手”。现在想知道这些同学中，有几对“旗鼓相当的对手”？同样一个人可能会和其他好几名同学结对。
输入格式

第一行一个正整数 NNN。

接下来 NNN 行，每行三个整数，其中第 iii 行表示第 iii 名同学的语文、数学、英语成绩。最先读入的同学编号为 1。
输出格式

输出一个个整数，表示“旗鼓相当的对手”的对数。
输入输出样例
输入 #1

3
90 90 90
85 95 90
80 100 91

输出 #1

2

import java.util.Scanner;

public class P5728 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[][] arr = new int[n][3];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < 3; j++) {
                arr[i][j] = sc.nextInt();
            }
        }
        System.out.println(f(n, arr));
    }

    public static int f(int m, int[][] a) {
        int flag = 0;
        for (int i = 0; i < m; i++) {
            for (int j = i + 1; j < m; j++) {
                if (Math.abs(a[i][0] - a[j][0]) <= 5 &&
                        Math.abs(a[i][1] - a[j][1]) <= 5 &&
                        Math.abs(a[i][2] - a[j][2]) <= 5 &&
                        Math.abs((a[i][0] + a[i][1] + a[i][2]) - (a[j][0] + a[j][1] + a[j][2])) <= 10) {
                    flag++;
                    //System.out.println("第"+i+","+"第"+j+"次");
                }
                else {
                    continue;
                }
            }
        }
        return flag;
    }
}