import java.util.ArrayList;
public class stations {
int[] label(int n, int k, int[] u, int[] v) {
ArrayList<Integer>[] G = new ArrayList[n];
for (int i = 0; i < G.length; i++) {
G[i] = new ArrayList<>();
}
for (int i = 0; i < u.length; i++) {
G[u[i]].add(v[i]);
G[v[i]].add(u[i]);
}
int maxSize = 0;
for (int i = 0; i < n; i++) {
maxSize = Math.max(maxSize, G[i].size());
}
int[] labels = new int[n];
if (maxSize > 0) {
for (int i = 0; i < n; i++) {
labels[i] = i;
}
return labels;
}
int leaf = -1;
for (int i = 0; i < n; i++) {
if (G[i].size() == 1) {
leaf = i;
break;
}
}
int cur = 0;
int node = leaf;
int prev = -1;
while (true) {
labels[node] = cur;
cur++;
boolean found = false;
for (int to : G[node]) {
if (to == prev) continue;
prev = node;
node = to;
found = true;
break;
}
if (!found) break;
}
return labels;
}
boolean isAncestor(int a, int b) {
if (a == b) return true;
while (a > 0) {
a = (a - 1) / 2;
if (a == b) return true;
}
return false;
}
int find_next_station(int s, int t, int[] c) {
boolean isBamboo = true;
if (c.length > 2) {
isBamboo = false;
}
else {
for (int nd : c) {
if (Math.abs(nd - s) > 1) isBamboo = false;
}
}
if (!isBamboo) {
if (!isAncestor(t, s)) {
return (s - 1) / 2;
}
for (int nd : c) {
if (nd < s) continue;
if (isAncestor(t, nd)) {
return nd;
}
}
} else {
for (int nd : c) {
if (s <= nd && nd <= t) return nd;
if (s >= nd && nd >= t) return nd;
}
}
return -1;
}
}
Compilation message
Note: stations.java uses unchecked or unsafe operations.
Note: Recompile with -Xlint:unchecked for details.
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1197 ms |
38472 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1201 ms |
37744 KB |
Output is correct |
2 |
Incorrect |
1251 ms |
35028 KB |
Wrong query response. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1280 ms |
36636 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1639 ms |
37796 KB |
Output is correct |
2 |
Incorrect |
1286 ms |
36000 KB |
Wrong query response. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1218 ms |
37028 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |