結果
| 問題 |
No.556 仁義なきサルたち
|
| ユーザー |
 Grenache
|
| 提出日時 | 2017-08-11 23:26:23 |
| 言語 | Java (openjdk 23) |
| 結果 |
AC
|
| 実行時間 | 147 ms / 2,000 ms |
| コード長 | 6,522 bytes |
| コンパイル時間 | 4,214 ms |
| コンパイル使用メモリ | 80,856 KB |
| 実行使用メモリ | 42,196 KB |
| 最終ジャッジ日時 | 2024-10-12 21:57:33 |
| 合計ジャッジ時間 | 6,929 ms |
|
ジャッジサーバーID （参考情報） |
judge3 / judge2 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| other | AC * 22 |
ソースコード
import java.io.*;
import java.util.*;
public class Main_yukicoder556 {
private static Scanner sc;
private static Printer pr;
private static void solve() {
int n = sc.nextInt();
int[] boss = new int[n];
for (int i = 0; i < n; i++) {
boss[i] = i;
}
UnionFind uf = new UnionFind(n);
List<List<Integer>> list = new ArrayList<>(n);
for (int i = 0; i < n; i++) {
List<Integer> tmp = new ArrayList<>();
tmp.add(i);
list.add(tmp);
}
int m = sc.nextInt();
for (int i = 0; i < m; i++) {
int a = sc.nextInt() - 1;
int b = sc.nextInt() - 1;
if (uf.same(a, b)) {
continue;
}
if (uf.count(a) == uf.count(b)) {
if (boss[a] < boss[b]) {
list.get(boss[a]).addAll(list.get(boss[b]));
for (int e : list.get(boss[b])) {
boss[e] = boss[a];
}
// list.set(boss[b], null);
} else {
list.get(boss[b]).addAll(list.get(boss[a]));
for (int e : list.get(boss[a])) {
boss[e] = boss[b];
}
// list.set(boss[a], null);
}
} else if (uf.count(a) > uf.count(b)) {
list.get(boss[a]).addAll(list.get(boss[b]));
for (int e : list.get(boss[b])) {
boss[e] = boss[a];
}
// list.set(boss[b], null);
} else {
list.get(boss[b]).addAll(list.get(boss[a]));
for (int e : list.get(boss[a])) {
boss[e] = boss[b];
}
// list.set(boss[a], null);
}
uf.unite(a, b);
}
for (int i = 0; i < n; i++) {
pr.println(boss[i] + 1);
}
}
@SuppressWarnings("unused")
private static class UnionFind {
int n;
// cnt : 異なる集合の数
int cnt;
// parent[x] : 0～n-1 の場合、要素xのroot要素
// : -1～-n の場合、自分自身がroot要素、
// -parent[x]でxを含む集合の要素数
int[] parent;
// rank[x] : 要素xが属する木の高さ
int[] rank;
UnionFind(int n) {
this.n = n;
cnt = n;
parent = new int[n];
Arrays.fill(parent, -1);
rank = new int[n];
Arrays.fill(rank, 0);
}
// xのrootを求める
int find(int x) {
if (parent[x] < 0) {
return x;
} else {
return parent[x] = find(parent[x]);
}
}
// xとyが同じ集合に属するのか
boolean same(int x, int y) {
return find(x) == find(y);
}
// xとyの属する集合を併合する
void unite(int x, int y) {
x = find(x);
y = find(y);
if (x == y) {
return;
}
cnt--;
// rankが大きい集合をrootにする
if (rank[x] > rank[y]) {
parent[x] += parent[y];
parent[y] = x;
} else {
parent[y] += parent[x];
parent[x] = y;
if (rank[x] == rank[y]) {
rank[y]++;
}
}
return;
}
// 要素xを含む集合の要素数
int count(int x) {
return -parent[find(x)];
}
// 異なる集合の数
int count() {
return cnt;
}
}
// ---------------------------------------------------
public static void main(String[] args) {
sc = new Scanner(System.in);
pr = new Printer(System.out);
solve();
pr.close();
sc.close();
}
@SuppressWarnings("unused")
private static class Scanner {
BufferedReader br;
Scanner (InputStream in) {
br = new BufferedReader(new InputStreamReader(in));
}
private boolean isPrintable(int ch) {
return ch >= '!' && ch <= '~';
}
private boolean isCRLF(int ch) {
return ch == '\n' || ch == '\r' || ch == -1;
}
private int nextPrintable() {
try {
int ch;
while (!isPrintable(ch = br.read())) {
if (ch == -1) {
throw new NoSuchElementException();
}
}
return ch;
} catch (IOException e) {
throw new NoSuchElementException();
}
}
String next() {
try {
int ch = nextPrintable();
StringBuilder sb = new StringBuilder();
do {
sb.appendCodePoint(ch);
} while (isPrintable(ch = br.read()));
return sb.toString();
} catch (IOException e) {
throw new NoSuchElementException();
}
}
int nextInt() {
try {
// parseInt from Integer.parseInt()
boolean negative = false;
int res = 0;
int limit = -Integer.MAX_VALUE;
int radix = 10;
int fc = nextPrintable();
if (fc < '0') {
if (fc == '-') {
negative = true;
limit = Integer.MIN_VALUE;
} else if (fc != '+') {
throw new NumberFormatException();
}
fc = br.read();
}
int multmin = limit / radix;
int ch = fc;
do {
int digit = ch - '0';
if (digit < 0 || digit >= radix) {
throw new NumberFormatException();
}
if (res < multmin) {
throw new NumberFormatException();
}
res *= radix;
if (res < limit + digit) {
throw new NumberFormatException();
}
res -= digit;
} while (isPrintable(ch = br.read()));
return negative ? res : -res;
} catch (IOException e) {
throw new NoSuchElementException();
}
}
long nextLong() {
try {
// parseLong from Long.parseLong()
boolean negative = false;
long res = 0;
long limit = -Long.MAX_VALUE;
int radix = 10;
int fc = nextPrintable();
if (fc < '0') {
if (fc == '-') {
negative = true;
limit = Long.MIN_VALUE;
} else if (fc != '+') {
throw new NumberFormatException();
}
fc = br.read();
}
long multmin = limit / radix;
int ch = fc;
do {
int digit = ch - '0';
if (digit < 0 || digit >= radix) {
throw new NumberFormatException();
}
if (res < multmin) {
throw new NumberFormatException();
}
res *= radix;
if (res < limit + digit) {
throw new NumberFormatException();
}
res -= digit;
} while (isPrintable(ch = br.read()));
return negative ? res : -res;
} catch (IOException e) {
throw new NoSuchElementException();
}
}
float nextFloat() {
return Float.parseFloat(next());
}
double nextDouble() {
return Double.parseDouble(next());
}
String nextLine() {
try {
int ch;
while (isCRLF(ch = br.read())) {
if (ch == -1) {
throw new NoSuchElementException();
}
}
StringBuilder sb = new StringBuilder();
do {
sb.appendCodePoint(ch);
} while (!isCRLF(ch = br.read()));
return sb.toString();
} catch (IOException e) {
throw new NoSuchElementException();
}
}
void close() {
try {
br.close();
} catch (IOException e) {
// throw new NoSuchElementException();
}
}
}
private static class Printer extends PrintWriter {
Printer(PrintStream out) {
super(out);
}
}
}