NYOJ 题目1022合纵连横

合纵连横

时间限制： 1000 ms  |  内存限制： 65535 KB

难度：3

描述

乱世天下，诸侯割据。每个诸侯王都有一片自己的领土。但是不是所有的诸侯王都是安分守己的，实力强大的诸侯国会设法吞并那些实力弱的，让自己的领土面积不断扩大。而实力弱的诸侯王为了不让自己的领土被吞并，他会联合一些其他同样弱小的诸侯国，组成联盟(联盟不止一个)，来共同抵抗那些强大的诸侯国。 强大的诸侯国为了瓦解这些联盟，派出了最优秀的间谍来离间他们，使一些诸侯国退出联盟。最开始，每个诸侯国是一个联盟。

有两种操作

1、U x y 表示x和y在同一个联盟。（0≤x，y<n）

2、D x   表示x退出联盟。

输入

多组测试数据
第一行两个数，n和m（1 ≤ n≤ 10^5, 1 ≤ m ≤10^5），分别表示诸侯国的个数和操作次数。
接下来有m行操作

输出

输出联盟的个数

样例输入

5 7U 0 1U 1 2U 0 3D 0U 1 4D 2U 0 210 1U 0 9

样例输出

Case #1: 2Case #2: 9

有篇<a href="http://blog.csdn.net/a915800048/article/details/41703865">文章</a> 讲的挺好的

不过数组要开大点, 记录每次mark的最远距离. 避免越界和超时

import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {

	static int MAX = 1000500;
	static int[] food = new int[MAX];
	static int[] box = new int[MAX];
	static boolean[] mark = new boolean[MAX];
	static int number = 1;
	
	public static void main(String[] args) {
		Scanner cin = new Scanner(new BufferedInputStream(System.in));
		while (cin.hasNext()) {
			int N = cin.nextInt();
			int M = cin.nextInt();

			int max = N;
			int empty = N;
			for (int i = 0; i < N; i++) {
				box[i] = food[i] = i;
			}

			for (int i = 0; i < M; i++) {
				String opt = cin.next();
				if ("U".equals(opt)) {
					int x = find(food[cin.nextInt()]);
					int y = find(food[cin.nextInt()]);
					if (x != y) {
						box[x] = y;
					}
				} else {
					food[cin.nextInt()] = empty;
					box[empty] = empty;
					empty++;
					max = Math.max(max, empty);
				}
			}

			for (int i = 0; i < max; i++) {
				mark[i] = false;
			}
			
			int result = 0;
			for(int i=0; i<N; i++) {
				if(!mark[find(food[i])]) {
					mark[find(food[i])] = true;
					result ++;
				}
			}
			System.out.printf("Case #%d: %d\n", number ++, result);
		}
		cin.close();
	}

	private static int find(int i) {
		int r = i;
		while (box[r] != r) {
			r = box[r];
		}

		int j;
		while (box[i] != r) {
			j = box[i];
			box[i] = r;
			i = j;
		}
		return r;
	}

}