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

题目描述

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

输入格式

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

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

输出格式

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

样例输入

3
90 90 90
85 95 90
80 100 91

样例输出

2

提示

数据保证，$2\le N\le 1000$ 且每科成绩为不超过 $150$ 的自然数。

思路分析

每个学生都有多个信息要存储，一个一维数组显然不能满足需求。
**存储方法1：**定义4个数组a、b、c、d分别存储学生的语文成绩、数学成绩、英语成绩和总分。
**存储方法2：**定义一个4列的二维数组，第0、1、2、3列分别保存每个学生的语文成绩、数学成绩、英语成绩和总分。
**存储方法3：**定义一个结构体，将学生的语文成绩、数学成绩、英语成绩和总分打包到一起，然后用结构题数组来存储。

两个不同的学生，在数组中的下标一定一个小一个大。为了不重复、不遗漏的比较所有两个不同的学生，可以使用双重嵌套循环： 外层循环变量i从1枚举到n-1，内层循环变量j从i+1枚举到n。

**参考代码1：**定义4个数组

#include<iostream>
using namespace std;
//a、b、c、d分别保存每个学生的语文成绩、数学成绩、英语成绩和总分
int a[1001], b[1001], c[1001], d[1001];
int main() {
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i] >> b[i] >> c[i];
		d[i] = a[i] + b[i] + c[i];
	}
	/*
	要枚举两个不同的学生，可以使用双重嵌套循环：
	外层循环变量i从1枚举到n-1，内层循环变量j从i+1枚举到n。
	这样可以保证i<j，并且能够不重复、不遗漏的比较所有两个不同的学生。
	*/
	int cnt = 0; //统计旗鼓相当的学生对儿数
	for (int i = 1; i < n; i++) { //枚举第1个学生
		for (int j = i + 1; j <= n; j++) { //枚举第2个学生
			bool fa = abs(a[i] - a[j]) <= 5;
			bool fb = abs(b[i] - b[j]) <= 5;
			bool fc = abs(c[i] - c[j]) <= 5;
			bool fd = abs(d[i] - d[j]) <= 10;
			if ( fa && fb && fc && fd )
				cnt++;
		}
	}
	cout << cnt;
	return 0;
}

**参考代码2：**定义二维数组

#include<iostream>
using namespace std;
int a[1001][4];//第0、1、2、3列分别保存每个学生的语文成绩、数学成绩、英语成绩和总分
int main() {
	int n;
	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];
	}
	/*
	要枚举两个不同的学生，可以使用双重嵌套循环：
	外层循环变量i从1枚举到n-1，内层循环变量j从i+1枚举到n。
	这样可以保证i<j，并且能够不重复、不遗漏的比较所有两个不同的学生。
	*/
	int cnt = 0; //统计旗鼓相当的学生对儿数
	for (int i = 1; i < n; i++) { //枚举第1个学生
		for (int j = i + 1; j <= n; j++) { //枚举第2个学生
			bool fa = abs(a[i][0] - a[j][0]) <= 5;
			bool fb = abs(a[i][1] - a[j][1]) <= 5;
			bool fc = abs(a[i][2] - a[j][2]) <= 5;
			bool ftot = abs(a[i][3] - a[j][3]) <= 10;
			if ( fa && fb && fc && ftot )
				cnt++;
		}
	}
	cout << cnt;
	return 0;
}

**参考代码3：**使用结构体数组

#include<iostream>
using namespace std;
struct stu {
	int a, b, c; //分别用来存储语文、数学和英语成绩
	int tot; //总分
} a[1001];
int main() {
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i].a >> a[i].b >> a[i].c;
		a[i].tot = a[i].a + a[i].b + a[i].c;
	}

	/*要枚举两个不同的学生，可以使用双重嵌套循环：
	外层循环变量i从1枚举到n-1，内层循环变量j从i+1枚举到n。
	这样可以保证i<j，并且能够不重复、不遗漏的比较所有两个不同的学生。
	*/
	int cnt = 0; //统计旗鼓相当的学生对儿数
	for (int i = 1; i < n; i++) { //枚举第1个学生
		for (int j = i + 1; j <= n; j++) { //枚举第2个学生
			bool fa = abs(a[i].a - a[j].a) <= 5;
			bool fb = abs(a[i].b - a[j].b) <= 5;
			bool fc = abs(a[i].c - a[j].c) <= 5;
			bool ftot = abs(a[i].tot - a[j].tot) <= 10;
			if ( fa && fb && fc && ftot )
				cnt++;
		}
	}
	cout << cnt;
	return 0;
}