D - New Friends(AtCoder Beginner Contest 350)
题目链接:
D - New Friends (atcoder.jp)
题目大意:
题目解析:
题目的大致意思: 假如A和B是朋友 B和C也是朋友 那么当A和C不是朋友的时候 可以通过B让A和C也成为朋友 问你增加了多少对的朋友关系
题目分析:
咱们可以从图论去考虑 当这一群是一个连通块 那么这一群点(人) 都是可以通过这个连通块去成为朋友的 那么假如这个连通块有N个人 那么就会有 N * (N - 1) / 2 条边(朋友关系) 那么全部的连通块减去之前的M条原有的朋友关系就是答案 注意开long 存取答案
那么问题来了 怎么去看他们是不是在一个连通块 并查集出手了
复习一下 并查集让点y到点x的连通下 那么就是p[find(y)] = find(x) 直接就过去了
代码:
import java.util.*;// 我认为这个题是并查集的题
public class D {public static int[] p = null; // 表示的是父亲public static void main(String[] args) {var sc = new Scanner(System.in);int n = sc.nextInt();int m = sc.nextInt();p = new int[n + 10];for(int i = 1; i <= n; i ++ ) {p[i] = i; // 刚开始的祖宗 都是自己自己}long ret = 0;int mm = m;var mp = new HashMap<Integer, Long>();var se = new TreeSet<Integer>();while(m -- != 0 ) {int x = sc.nextInt();int y = sc.nextInt();// 这里理解为y加到x的节点下面p[find(y)] = find(x); // 该不会是这里出问题了吧}for(int i = 1; i <= n; i ++ ) {int x = p[find(i)];if(!mp.containsKey(x)) {mp.put(x, 0l);}long t = mp.get(x) + 1;mp.put(x, t);se.add(x); // 这里面存的是 都是祖宗}for(int i : se) {//System.out.print("i = " + i + "\n");//System.out.print("x = " + mp.get(i) + "\n");long tt = mp.get(i);long t2 = mp.get(i) - 1;ret += tt * t2 / 2;//System.out.print("ret = " + ret + "\n");}ret -= mm;System.out.print(ret);}public static int find(int x) {if(x != p[x])p[x] = find(p[x]);return p[x];}
}
运行的结果:
因为long问题和并查集认祖宗的问题 出现了几次wa