AcWing 第 143 场周赛 B.(5471) 数对推理 (Java)

AcWing 第 143 场周赛 B.(5471) 数对推理 (Java)

比赛链接：第 143 场周赛

B题传送门：B.(5471) 数对推理

题目：5471. 数对推理

奶牛贝茜和奶牛贝蒂各有一个整数数对。

每个数对都包含两个 1∼9之间的不同整数。

这两个数对恰好包含一个公共数，即恰好有一个整数同时包含于这两个数对。

初始时，贝茜和贝蒂都只知道自己拥有的数对，而不清楚对方拥有的数对。

它们希望通过网络交流来获知彼此数对的公共数。

但是，它们清楚所有交流信息一定会被农夫约翰截获，且约翰也知道这两个数对恰好包含一个公共数。

为了让截获到交流信息的约翰无法获知这个公共数，它们决定采用如下方式加密信息。

贝茜会准备 n 个两两不同的数对，这些数对均包含两个 1∼9之间的不同整数，其中一个数对恰好是它拥有的数对，它会将这些数对全部发送给贝蒂。

贝蒂会准备 m个两两不同的数对，这些数对均包含两个 1∼9之间的不同整数，其中一个数对恰好是它拥有的数对，它会将这些数对全部发送给贝茜。

当然，约翰会将这些数对（以及是谁发送的）全部获知。

请你对给定信息进行判断，并按要求输出：

• 如果这些信息足以令约翰准确地推断出公共数字，则输出这个公共数字。
• 如果这些信息不足以令约翰准确地推断出公共数字，但是不论两头奶牛各自拥有哪个数对，都足以令两头奶牛都准确地推断出公共数字，则输出 0。
• 以上都不是，则输出 -1。

输入格式
第一行包含两个整数 n,m。

第二行包含 n个两两不同的数对，这些数对均包含两个 1∼9之间的不同整数，表示贝茜准备的数对。

第三行包含 m个两两不同的数对，这些数对均包含两个 1∼9之间的不同整数，表示贝蒂准备的数对。

在本题中，(1,2)和 (2,1)视为同一数对。

数据保证双方拥有的数对包含于各自给出的数对当中，且双方拥有的数对恰好包含一个公共数。

输出格式

按照题目要求，输出公共数字或 0 或 -1。

数据范围

前 4个测试点满足 1≤n,m≤3。
所有测试点满足 1≤n,m≤12。

输入样例1：

2 2
1 2 3 4
1 5 3 4

输出样例1：

1

输入样例2：

2 2
1 2 3 4
1 5 6 4

输出样例2：

0

输入样例3：

2 3
1 2 4 5
1 2 1 3 2 3

输出样例3：

-1

分析：

get_common()函数用于判断两个数对的公共数是什么。如果两个数对完全相同，令公共数为0；如果其中一个数对的某个数在另一个数对中出现，则该数为公共数；否则，公共数为0。

check_f()函数用于判断农夫能否准确地推断出公共数字。如果农夫能推断出唯一的公共数字，则返回该数字；否则返回0。

check_cow()函数用于判断两头奶牛能否准确地推断出公共数字。如果对于每一个奶牛，只要能推断出的公共数字都是唯一的，就返回true，否则返回false。

根据以上函数的返回值来输出结果。如果约翰能推断出唯一的公共数字，则输出该数字；如果两头奶牛都能准确地推断出公共数字，则输出0；否则输出-1。

代码：

import java.util.*;

public class Main{
	static int [][] a;
	static int [][] b;
	static int n,m;
	static int [][] g;
	
	// 判断两个数对的公共数
	public static int get_common(int[] a,int[] b) { 
		// 两个数对相同，令公共数为0
		if(a[0]==b[0]&&a[1]==b[1]) return 0;
		if(a[0]==b[0]||a[0]==b[1]) return a[0];
		if(a[1]==b[0]||a[1]==b[1]) return a[1];
		return 0;
	}
	
	// 判断农夫推断公共数是否唯一，并返回公共数
	public static int check_f() {
		int common = -1;
		for(int i = 0;i < n;i++) {
			for(int j = 0;j < m;j++) {
				if(g[i][j]>0) {
				    if(common!=-1&&common!=g[i][j]) return 0;
				    common = g[i][j];
				}
			}
		}
		return common;
	}
	
	// 判断两个奶牛的能否推断公共数
	public static boolean check_cow() {
		for(int i = 0;i < n;i++) {
			for(int j = 0;j < m;j++) {
				if(g[i][j]>0) {
					for(int k = 0;k < m;k++) {
						if(g[i][k]>0&&g[i][k]!=g[i][j])
							return false;
					}
					for(int k = 0;k < n;k++) {
						if(g[k][j]>0&&g[k][j]!=g[i][j])
							return false;
					}
				}
			}
		}
		return true;
	}
	
    public static void main(String[] args) {
    	Scanner sc = new Scanner(System.in);
    	n = sc.nextInt();m = sc.nextInt();
    	a = new int [n][2];
    	b = new int [m][2];
    	g = new int [n][m];
    	for(int i = 0;i < n;i++) {
    		a[i][0] = sc.nextInt();a[i][1] = sc.nextInt();
    		Arrays.sort(a[i]);
    	}
    	for(int i = 0;i < m;i++) {
    		b[i][0] = sc.nextInt();b[i][1] = sc.nextInt();
    		Arrays.sort(b[i]);
    	}
    	for(int i = 0;i < n;i++) {
    		for(int j = 0;j < m;j++) {
    			g[i][j] = get_common(a[i],b[j]);
    		}
    	}
    	
    	int t = check_f();
    	if(t>0) {
    		System.out.println(t);
    	}else if(check_cow()) {
    		System.out.println(0);
    	}else {
    		System.out.println(-1);
    	}
    	
    	sc.close();
    }
}