藍橋杯 歷屆試題 PREV-12 危險係數 並查集找割點 Java

歷屆試題 危險係數  

時間限制：1.0s   內存限制：256.0MB

      

問題描述

抗日戰爭時期，冀中平原的地道戰曾發揮重要作用。

地道的多個站點間有通道連接，形成了龐大的網絡。但也有隱患，當敵人發現了某個站點後，其它站點間可能因此會失去聯繫。

我們來定義一個危險係數DF(x,y)：

對於兩個站點x和y (x != y), 如果能找到一個站點z，當z被敵人破壞後，x和y不連通，那麼我們稱z爲關於x,y的關鍵點。相應的，對於任意一對站點x和y，危險係數DF(x,y)就表示爲這兩點之間的關鍵點個數。

本題的任務是：已知網絡結構，求兩站點之間的危險係數。

輸入格式

輸入數據第一行包含2個整數n(2 <= n <= 1000), m(0 <= m <= 2000),分別代表站點數，通道數；

接下來m行，每行兩個整數 u,v (1 <= u, v <= n; u != v)代表一條通道；

最後1行，兩個數u,v，代表詢問兩點之間的危險係數DF(u, v)。

輸出格式

一個整數，如果詢問的兩點不連通則輸出-1.

樣例輸入

7 6
1 3
2 3
3 4
3 5
4 5
5 6
1 6

樣例輸出

2

並查集有個細節總是喜歡搞錯，記錄一下

Merge(3, 6)

import java.util.Scanner;

public class Main {
	static int n, m, start, end, lene, cnt;
	static int[] f = new int[1005];
	static Edge[] edges = new Edge[2005];
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		m = in.nextInt();
		lene = 0;
		for (int i = 0; i < m; i++) {
			int u, v;
			u = in.nextInt();
			v = in.nextInt();
			Edge e = new Edge(u, v);
			edges[lene++] = e;
		}
		start = in.nextInt();
		end = in.nextInt();
		
		if (solve()) {
			System.out.println(cnt);
		} else {
			System.out.println(-1);
		}
		
		in.close();
	}
	
	static void initf() {
		for (int i = 1; i <= n; i++) {
			f[i] = i;
		}
	}
	
	static int getf(int x) {
		return f[x] == x ? x : (f[x] = getf(f[x]));
	}
	
	static void Merge(int x, int y) {
		int u = getf(x);
		int v = getf(y);
		if (u != v) {
			if (u < v) {
				f[v] = u;
			} else {
				f[u] = v;
			}
		}
	}
	
	static boolean solve() {
		initf();
		for (int i = 0; i < lene; i++) {
			Merge(edges[i].u, edges[i].v);
		}
		//debug
		//if(f[start] != f[end]) --> if (getf(start) != getf(end))
		if (getf(start) != getf(end)) return false;
		
		cnt = 0;
		//遍歷可能被破壞的i點
		for (int i = 1; i <= n; i++) {
			if (i != start && i != end) {
				initf();
				for (int j = 0; j < lene; j++) {
					int u = edges[j].u, v = edges[j].v;
					if (u != i && v != i) {
						Merge(u, v);
					}
				}
				//debug
				//if(f[start] != f[end]) --> if (getf(start) != getf(end))
				if (getf(start) != getf(end)) {
					cnt++;
				}
			}
		}
		return true;
	}
}

class Edge {
	int u, v;
	public Edge() {}
	public Edge(int u, int v) {
		this.u = u;
		this.v = v;
	}
}