1. 题目描述

给定五个 0 ~ 9 范围内的整数 a1, a2, a3, a4, a5。如果能从五个整数中选出三个并且这三个整数的和为10 的倍数(包括 0),那么这五个整数的权值即为剩下两个没被选出来的整数的和对 10 取余的结果,显然如果有多个三元组满⾜和是 10 的倍数,剩下两个数之和对 10 取余的结果都是相同的;如果选不出这样三个整数,则这五个整数的权值为 -1。现在给定 T 组数据,每组数据包含五个 0~9 范围内的整数,分别求这 T 组数据中五个整数的权值。

输入描述

输出描述

输入样例

4
1 0 0 1 0
1 0 0 8 6
3 4 5 6 7
4 5 6 7 8

输出样例

2
-1
-1
0

2. 我的尝试

对于每组数据,记五数之和对10取模的结果为mod,只需穷举该组数据中所有的二元组,若存在某个二元组两数之和对10取模后的结果等于mod,则该组数据权重即为mod。若没有任意一个二元组匹配成功,则权重为-1

每组数据只需穷举 C 5 2 = 10 C^2_5 = 10 C52=10 次,总的时间复杂度为线性的。

#include <bits/stdc++.h>

using namespace std;

int main() {
	int n;
	int a[5];
	bool flag = false;
	int mod;
	
	cin >> n;
	for (int i = 0; i < n; i ++){
	    flag = false;
	    mod = 0;
	    
	    for (int j = 0; j < 5; j ++){
	        scanf("%d", &a[j]);
	        mod = (mod + a[j]) % 10;
	    }
	        
	    for (int j = 0; j < 5; j ++)
	        for (int k = 0; k < j; k ++)
	            if ((a[j] + a[k]) % 10 == mod) \
	                flag = true;
	   
	   if (flag) printf("%d\n", mod);
	   else printf("-1\n");
	}
	    
	return 0;
}
03-14 10:30