提出 #74168204

---

ソースコード 拡げる

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) {
        InputReader reader = new InputReader(System.in);
        PrintWriter writer = new PrintWriter(System.out, false);
        int N = reader.nextInt();
        int K = reader.nextInt();
        int[] A = new int[N];
        for (int i = 0; i < N; i++) {
            A[i] = reader.nextInt();
        }
        writer.println(atMost(A, K) - atMost(A, K - 1));
        writer.close();
        System.exit(0);
    }

    private static long atMost(int[] A, int K) {
        if (K < 0) return 0;
        long result = 0;
        TreeMap<Integer, Integer> tempCount = new TreeMap<>();
        for (int r = 0, l = 0; r < A.length; r++) {
            tempCount.put(A[r], tempCount.getOrDefault(A[r], 0) + 1);
            while (tempCount.lastKey() - tempCount.firstKey() > K) {
                tempCount.put(A[l], tempCount.get(A[l]) - 1);
                tempCount.remove(A[l++], 0);
            }
            result += r - l + 1;
        }
        return result;
    }

    static class InputReader {
        public BufferedReader reader;
        public StringTokenizer tokenizer;
        public InputReader(InputStream stream) {
            reader = new BufferedReader(new InputStreamReader(stream), 32768);
            tokenizer = null;
        }
        public String next() {
            while (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }
        public int nextInt() {
            return Integer.parseInt(next());
        }
        public long nextLong() {
            return Long.parseLong(next());
        }
        public double nextDouble() {
            return Double.parseDouble(next());
        }
        public String nextLine() {
            String str = "";
            try {
                str = reader.readLine();
            } catch (IOException e) {
                e.printStackTrace();
            }
            return str;
        }
    }
}

提出情報

ジャッジ結果