結果

問題 No.1170 Never Want to Walk
ユーザー tentententen
提出日時 2020-08-17 11:19:38
言語 Java21
(openjdk 21)
結果
AC  
実行時間 953 ms / 2,000 ms
コード長 2,052 bytes
コンパイル時間 4,684 ms
コンパイル使用メモリ 77,448 KB
実行使用メモリ 76,584 KB
最終ジャッジ日時 2024-10-11 10:55:13
合計ジャッジ時間 20,905 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 134 ms
41,308 KB
testcase_01 AC 133 ms
41,060 KB
testcase_02 AC 134 ms
41,160 KB
testcase_03 AC 134 ms
40,828 KB
testcase_04 AC 137 ms
41,164 KB
testcase_05 AC 135 ms
41,140 KB
testcase_06 AC 136 ms
41,076 KB
testcase_07 AC 135 ms
41,260 KB
testcase_08 AC 132 ms
40,908 KB
testcase_09 AC 133 ms
41,012 KB
testcase_10 AC 133 ms
40,916 KB
testcase_11 AC 134 ms
41,072 KB
testcase_12 AC 197 ms
42,008 KB
testcase_13 AC 199 ms
42,432 KB
testcase_14 AC 195 ms
42,348 KB
testcase_15 AC 200 ms
42,052 KB
testcase_16 AC 197 ms
42,396 KB
testcase_17 AC 197 ms
42,624 KB
testcase_18 AC 193 ms
42,796 KB
testcase_19 AC 200 ms
42,156 KB
testcase_20 AC 200 ms
42,276 KB
testcase_21 AC 188 ms
42,328 KB
testcase_22 AC 192 ms
42,348 KB
testcase_23 AC 199 ms
42,192 KB
testcase_24 AC 193 ms
42,264 KB
testcase_25 AC 200 ms
42,236 KB
testcase_26 AC 199 ms
42,476 KB
testcase_27 AC 909 ms
59,384 KB
testcase_28 AC 886 ms
58,848 KB
testcase_29 AC 909 ms
59,496 KB
testcase_30 AC 918 ms
58,756 KB
testcase_31 AC 918 ms
59,668 KB
testcase_32 AC 932 ms
75,988 KB
testcase_33 AC 934 ms
65,736 KB
testcase_34 AC 953 ms
76,584 KB
testcase_35 AC 953 ms
73,140 KB
testcase_36 AC 920 ms
74,436 KB
testcase_37 AC 932 ms
72,236 KB
testcase_38 AC 912 ms
72,848 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int a = sc.nextInt();
        int b = sc.nextInt();
        int[] arr = new int[n + 1];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        arr[n] = Integer.MAX_VALUE;
        int left = 0;
        int right = 0;
        int last = 0;
        UnionFindTree uft = new UnionFindTree(n);
        for (int i = 0; i < n; i++) {
            while (arr[left] - arr[i] < a) {
                left++;
            }
            if (left == n) {
                break;
            }
            while(arr[right] - arr[i] <= b) {
                right++;
            }
            for (int j = Math.max(left, last); j < right; j++) {
                uft.unite(i, j);
            }
            last = right - 1;
        }
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < n; i++) {
            sb.append(uft.getCount(i)).append("\n");
        }
        System.out.print(sb);
   }
   
   static class UnionFindTree {
       int[] parents;
       int[] counts;
       
       public UnionFindTree(int size) {
           parents = new int[size];
           counts = new int[size];
           for (int i = 0; i < size; i++) {
               parents[i] = i;
               counts[i] = 1;
           }
       }
       
       public int find(int x) {
           if (x == parents[x]) {
               return x;
           } else {
               return parents[x] = find(parents[x]);
           }
       }
       
       public boolean same(int x, int y) {
           return find(x) == find(y);
       }
       
       public void unite(int x, int y) {
           if (!same(x, y)) {
               int xx =find(x);
               int yy = find(y);
               parents[xx] = yy;
               counts[yy] += counts[xx];
           }
       }
       
       public int getCount(int x) {
           return counts[find(x)];
       }
   }
}
0