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

题目描述

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

输入格式

第一行一个正整数 $N$。

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

输出格式

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

样例 #1

样例输入 #1

3
90 90 90
85 95 90
80 100 91

样例输出 #1

2

我的做法：

#include<iostream>
#include<cmath>
using namespace std;
int main() {
    int n, a[1010][6];
    int ans;
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> a[i][0] >> a[i][1] >> a[i][2];
        a[i][3] = a[i][0] + a[i][1] + a[i][2];
    }
    for(int i = 1; i <= n; i++){
        for(int j = i + 1; j <= n; j++)
            if(abs(a[i][0] - a[j][0]) <= 5 &&
                   abs(a[i][1] - a[i][1]) <= 5 &&
                   abs(a[i][2] - a[j][2]) <= 5 &&
                   abs(a[i][3] - a[j][3]) <= 10)
                ans++;
    }
    cout << ans;
    return 0;
}