蓝桥杯 整理玩具【第九届】【决赛】【B组】

蓝桥杯 整理玩具【第九届】【决赛】【B组】

资源限制
时间限制：1.0s 内存限制：256.0MB
　　小明有一套玩具，一共包含NxM个部件。这些部件摆放在一个包含NxM个小格子的玩具盒中，每个小格子中恰好摆放一个部件。

每一个部件上标记有一个0~9的整数，有可能有多个部件标记相同的整数。

小明对玩具的摆放有特殊的要求：标记相同整数的部件必须摆在一起，组成一个矩形形状。

如以下摆放是满足要求的：

00022
00033
44444

12244
12244
12233

01234
56789
以下摆放不满足要求：

11122
11122
33311

111111
122221
122221
111111

11122
11113
33333

给出一种摆放方式，请你判断是否符合小明的要求。
输入格式
　　----
　　输入包含多组数据。
　　第一行包含一个整数T，代表数据组数。 (1 <= T <= 10)
　　以下包含T组数据。
　　每组数据第一行包含两个整数N和M。 (1 <= N, M <= 10)
　　以下包含N行M列的矩阵，代表摆放方式。
输出格式
　　—
　　对于每组数据，输出YES或者NO代表是否符合小明的要求。
样例输入
3
3 5
00022
00033
44444
3 5
11122
11122
33311
2 5
01234
56789
样例输出
YES
NO
YES
资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗 < 1000ms

记录每一个数字出现的头和尾，计算该矩形块内数字个数，然后将所有数字的覆盖的面积相加，如果整数不是摆在一起则有面积重复。如果符合要求则面积相加等于整个数组的总数。

import java.util.*;


public class Main {
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		int N=in.nextInt();
		List<int[][]> list=new ArrayList<int[][]>();
		for(int i=0;i<N;i++) {
			int n,m;
			n=in.nextInt();
			m=in.nextInt();
			int[][] a=new int[n][m];
			for(int j=0;j<n;j++) {
				String s=in.next();
				for(int k=0;k<m;k++) {
					a[j][k]=Integer.parseInt(s.charAt(k)+"");
				}
			}
			list.add(a);
		}
		for(int i=0;i<N;i++) {
			int[][] b=list.get(i);
			isRectangle(b);
		}
	}
	public static void isRectangle(int[][] a) {
		re[] r = new re[10];
		for(int i=0;i<10;i++) {
			r[i]=new re();
		}
		for(int i=0;i<a.length;i++) {
			for(int j=0;j<a[0].length;j++) {
				r[a[i][j]].setRe(i, j);
			}
		}
		int res=0;
		for(int i=0;i<10;i++) {
				res=res+r[i].getA();
		}
		if(res==a.length*a[0].length) {
			System.out.println("YES");
		}
		else
			System.out.println("NO");
	}
}
class re{
	int minI=Integer.MAX_VALUE,minJ=Integer.MAX_VALUE,maxI=Integer.MIN_VALUE,maxJ=Integer.MIN_VALUE;
	public re() {
		
	}
	public void setRe(int i,int j) {
		if(i<=minI&&j<=minJ) {
			minI=i;minJ=j;
		}
		if(i>=maxI&&j>=maxJ) {
			maxI=i;maxJ=j;
		}
	}
	public int getA() {
		if(maxI!=Integer.MIN_VALUE)
		return (maxI-minI+1)*(maxJ-minJ+1);
		else
			return 0;
	}
}