P5728 旗鼓相当的对手

题目描述

现有 𝑁N 名同学参加了期末考试，并且获得了每名同学的信息：语文、数学、英语成绩（均为不超过 150150 的自然数）。如果某对学生 〈𝑖,𝑗〉〈i,j〉 的每一科成绩的分差都不大于 55，且总分分差不大于 1010，那么这对学生就是“旗鼓相当的对手”。现在想知道这些同学中，有几对“旗鼓相当的对手”？同样一个人可能会和其他好几名同学结对。

输入格式

第一行一个正整数 𝑁N。

接下来 𝑁N 行，每行三个整数，其中第 𝑖i 行表示第 𝑖i 名同学的语文、数学、英语成绩。最先读入的同学编号为 11。

输出格式

输出一个整数，表示“旗鼓相当的对手”的对数。

输入输出样例

输入 #1复制

3
90 90 90
85 95 90
80 100 91

输出 #1复制

2

说明/提示

数据保证，2≤𝑁≤10002≤N≤1000 且每科成绩为不超过 150150 的自然数。

代码：

#include<iostream>
#include<cmath>
#define N 1001
using namespace std;
int main() {
	int arr[N][4];
	int n; cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> arr[i][0] >> arr[i][1] >> arr[i][2];
		arr[i][3] = arr[i][0] + arr[i][1] + arr[i][2];
	}
	int count = 0;
	for (int i = 0; i < n - 1; i++) {
		for (int j = i + 1; j < n; j++) {
			if (abs(arr[i][0] - arr[j][0]) <= 5 &&
				abs(arr[i][1] - arr[j][1]) <= 5 &&
				abs(arr[i][2] - arr[j][2]) <= 5 &&
				abs(arr[i][3] - arr[j][3]) <= 10) {
				count++;
			}
		}
	}
	cout << count << endl;
	return 0;
}