-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathKnightTournament.java
More file actions
90 lines (77 loc) · 2.79 KB
/
KnightTournament.java
File metadata and controls
90 lines (77 loc) · 2.79 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import java.util.Scanner;
class SegmentTree {
private int[] tree;
private int[] defeatedBy;
private int n;
public SegmentTree(int size) {
n = size;
tree = new int[4 * n]; // Segment tree size
defeatedBy = new int[n + 1]; // To store who defeated whom
}
// Build the segment tree initially with zeros
public void build() {
build(1, 1, n);
}
private void build(int node, int start, int end) {
if (start == end) {
tree[node] = 0; // No winner initially
} else {
int mid = (start + end) / 2;
build(2 * node, start, mid);
build(2 * node + 1, mid + 1, end);
tree[node] = 0; // Initialize to zero
}
}
// Update the segment tree and defeatedBy array
public void update(int l, int r, int winner) {
update(1, 1, n, l, r, winner);
}
private void update(int node, int start, int end, int l, int r, int winner) {
// Update range
if (start > r || end < l) {
return; // No overlap
}
if (start >= l && end <= r) {
for (int i = start; i <= end; i++) {
if (defeatedBy[i] == 0 && i != winner) {
defeatedBy[i] = winner; // Mark defeated by winner
}
}
tree[node] = winner; // Set the winner in this segment
return; // Full overlap
}
// Partial overlap
int mid = (start + end) / 2;
update(2 * node, start, mid, l, r, winner);
update(2 * node + 1, mid + 1, end, l, r, winner);
}
// Get defeatedBy information
public int[] getDefeatedBy() {
return defeatedBy;
}
}
public class KnightTournament {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt(); // number of knights
int m = sc.nextInt(); // number of fights
SegmentTree segmentTree = new SegmentTree(n);
segmentTree.build();
// Process each fight
for (int i = 0; i < m; i++) {
int l = sc.nextInt(); // start index of knights in this fight
int r = sc.nextInt(); // end index of knights in this fight
int x = sc.nextInt(); // winner of this fight
segmentTree.update(l, r, x); // Update the segment tree
}
// The last winner is not defeated by anyone
int lastWinner = sc.nextInt(); // The last winner from the last fight
segmentTree.getDefeatedBy()[lastWinner] = 0; // Set last winner's value to 0
// Output the results for knights 1 to n
int[] result = segmentTree.getDefeatedBy();
for (int i = 1; i <= n; i++) {
System.out.print(result[i] + " ");
}
sc.close();
}
}