poj 3467 Cross Counting

題意:給出一個M*N 的矩陣,矩陣中每個元素都有一種顏色(1 ≤ N, M, C ≤ 100),10000條查詢,更改某個元素的顏色或查詢某種顏色的corss有多少個。

We say there exists a cross of size k centered at the cell (x,y)iff all cells lying in the x-th row or the y-th column and within a distance of k from (x,y) share the same color. if two crosses have the same center but different sizes we consider they are distinct

解法:由於每種顏色之間互不相關,因此可以分顏色考慮,將矩陣變爲100個01矩陣,這樣可以統計出初始時每種顏色有多少個cross。

當某個元素(i,j)發生改變時,隻影響i行或j列爲中心的cross,因此只需逐個考慮每個中心的cross的變化,每次查詢只需更改200的元素的cross數目。怎麼快速統計出以每個元素爲中心的cross數目呢?根據定義,只需記錄01矩陣中每個元素向上下左右四個方向連續1的個數,取最小值即可,當一個元素從1變爲0或從0變爲1,只需改變同行元素的左右方向連續1的個數和同列元素的上下方向連續1的個數,並記錄cross數目增加減少的數量即可。

貌似線段樹也可以維護,沒想到這麼維護,所以就暫且認爲複雜度比這個高了(囧)。。。

import java.util.Scanner;

public class CountCross3467 {
	int maxn = 110;
	class Matrix {
		int map[][] = new int[maxn][maxn];
		int left[][] = new int[maxn][maxn], right[][] = new int[maxn][maxn],
				up[][] = new int[maxn][maxn], down[][] = new int[maxn][maxn];
		int sum, n, m;
        Matrix(int n,int m){
        	this.n=n;
        	this.m=m;
        }
		void init(int i, int j) {
			map[i][j] = 1;
			left[i][j] = right[i][j] = j;
			up[i][j] = down[i][j] = i;
		}
		void count() {
			for (int i = 1; i <= n; i++) {
				for (int j = 1; j <= m; j++)
					if (map[i][j] == 1 && map[i][j - 1] == 1)
						left[i][j] = left[i][j - 1];					
				for (int j = m; j > 0; j--)
					if (map[i][j] == 1 && map[i][j + 1] == 1)
						right[i][j] = right[i][j + 1];	
			}
			
			for (int i = 1; i <= m; i++) {
				for (int j = 1; j <= n; j++)
					if (map[j][i] == 1 && map[j - 1][i] == 1)
						up[j][i] = up[j - 1][i];
				for (int j = n; j > 0; j--)
					if (map[j][i] == 1 && map[j + 1][i] == 1)
						down[j][i] = down[j + 1][i];
			}
			
			sum = 0;
			for (int i = 1; i <= n; i++)
				for (int j = 1; j <= m; j++) 
					sum += cross(i, j);				
		}
		int cross(int i, int j) {
			if(map[i][j]==0)
				return 0;
			int a = Math.min(i - up[i][j], down[i][j] - i);
			int b = Math.min(j-left[i][j], right[i][j] - j);
			return Math.min(a, b);
		}
		void update1(int a, int b) {
			// 0-->1
			map[a][b] = 1;
			left[a][b] = b;
			if (map[a][b - 1] == 1)
				left[a][b] = left[a][b - 1];
			right[a][b] = b;
			if (map[a][b + 1] == 1)
				right[a][b] = right[a][b + 1];
			up[a][b] = a;
			if (map[a - 1][b] == 1)
				up[a][b] = up[a - 1][b];
			down[a][b] = a;
			if (map[a + 1][b] == 1)
				down[a][b] = down[a + 1][b];
			sum+=cross(a,b);
			for (int i = 1; i < b; i++) {
				if (right[a][i] != b - 1)
					continue;
				int temp = cross(a, i);
				right[a][i] = right[a][b];
				sum += cross(a, i) - temp;
			}
			for (int i = b + 1; i <= m; i++) {
				if (left[a][i] != b + 1)
					continue;
				int temp = cross(a, i);
				left[a][i] = left[a][b];
				sum += cross(a, i) - temp;
			}
			for(int i=1;i<a;i++){
				if(down[i][b]!=a-1)
					continue;
				int temp=cross(i,b);
				down[i][b]=down[a][b];
				sum += cross(i,b) - temp;
			}
			for(int i=a+1;i<=n;i++){
				if(up[i][b]!=a+1)
					continue;
				int temp=cross(i,b);
				up[i][b]=up[a][b];
				sum += cross(i,b) - temp;
			}
		}
		void update0(int a, int b) {
			// 1-->0
			sum-=cross(a,b);
			map[a][b] = 0;
			left[a][b] =right[a][b]=up[a][b]=down[a][b]=0;
			for (int i = 1; i < b; i++) {
				if (map[a][i]==0||right[a][i]<b)
					continue;
				int temp = cross(a, i);
				right[a][i] =b-1;
				sum -= temp-cross(a, i);
			}
			for (int i = b + 1; i <= m; i++) {
				if (map[a][i]==0||left[a][i]>b)
					continue;
				int temp = cross(a, i);
				left[a][i] = b+1;
				sum -= temp-cross(a, i);
			}
			for(int i=1;i<a;i++){
				if(map[i][b]==0||down[i][b]<a)
					continue;
				int temp=cross(i,b);
				down[i][b]=a-1;
				sum-= temp-cross(i,b);
			}
			for(int i=a+1;i<=n;i++){
				if(map[i][b]==0||up[i][b]>a)
					continue;
				int temp=cross(i,b);
				up[i][b]=a+1;
				sum -= temp-cross(i,b);
			}
		}
	}
	Scanner scan=new Scanner(System.in);

	void run(){
		int n=scan.nextInt();
		int m=scan.nextInt();
		int c=scan.nextInt();
		int q=scan.nextInt();
		int map[][]=new int[n+1][m+1];
		Matrix mat[]=new Matrix[c+1];
		for(int i=1;i<=c;i++)
			mat[i]=new Matrix(n,m);
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++){
				map[i][j]=scan.nextInt();
				mat[map[i][j]].init(i, j);
			}
		for(int i=1;i<=c;i++)
			mat[i].count();
		
		while(q-->0){
			String s=scan.next();
			if(s.charAt(0)=='Q'){
				int v=scan.nextInt();
				System.out.println(mat[v].sum);
			}
			else{
				int i=scan.nextInt();
				int j=scan.nextInt();
				int v=scan.nextInt();
				int t=map[i][j];
				map[i][j]=v;
				mat[t].update0(i, j);
				mat[v].update1(i, j);
			}
		}
	}

	public static void main(String[] args) {
		new CountCross3467().run();
	}
}


發佈了75 篇原創文章 · 獲贊 15 · 訪問量 11萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章