結果

問題 No.877 Range ReLU Query
ユーザー kusomushikusomushi
提出日時 2019-09-09 00:23:20
言語 Java21
(openjdk 21)
結果
AC  
実行時間 1,231 ms / 2,000 ms
コード長 7,370 bytes
コンパイル時間 2,728 ms
コンパイル使用メモリ 91,068 KB
実行使用メモリ 75,600 KB
最終ジャッジ日時 2024-04-25 22:13:31
合計ジャッジ時間 16,749 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 78 ms
51,352 KB
testcase_01 AC 140 ms
52,700 KB
testcase_02 AC 138 ms
52,832 KB
testcase_03 AC 139 ms
52,716 KB
testcase_04 AC 87 ms
51,532 KB
testcase_05 AC 113 ms
52,268 KB
testcase_06 AC 117 ms
52,792 KB
testcase_07 AC 103 ms
52,216 KB
testcase_08 AC 140 ms
52,832 KB
testcase_09 AC 87 ms
51,532 KB
testcase_10 AC 130 ms
52,844 KB
testcase_11 AC 1,135 ms
75,600 KB
testcase_12 AC 1,070 ms
74,944 KB
testcase_13 AC 997 ms
66,304 KB
testcase_14 AC 952 ms
67,144 KB
testcase_15 AC 1,231 ms
73,348 KB
testcase_16 AC 1,157 ms
74,360 KB
testcase_17 AC 1,146 ms
73,564 KB
testcase_18 AC 1,160 ms
73,556 KB
testcase_19 AC 1,131 ms
73,112 KB
testcase_20 AC 1,199 ms
73,976 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.*;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringJoiner;
import java.util.StringTokenizer;
import java.util.function.Function;

public class Main {

    public static void main(String[] args) {
        FastScanner sc = new FastScanner(System.in);
        PrintWriter pw = new PrintWriter(System.out);

        int N = sc.nextInt();
        int Q = sc.nextInt();
        int[] A = sc.nextIntArray(N);
        int[][] events = new int[N+Q][];
        int idx = 0;
        for (int i = 0; i < N; i++) {
            events[idx++] = new int[]{2, A[i], i}; // queryより後
        }
        for (int q = 0; q < Q; q++) {
            int type = sc.nextInt(); // 使わない
            int l = sc.nextInt()-1;
            int r = sc.nextInt()-1;
            int x = sc.nextInt();
            events[idx++] = new int[]{1, x, l, r, q};
        }

        // x|aの降順
        // queryのほうが後(0になるのでどっちでもよいかも)
        Arrays.sort(events, Comparator.<int[]>comparingInt(e -> -e[1]).thenComparingInt(e -> e[1]));

        RangeGet values = new RangeGet(N, RangeGet.SUM);
        RangeGet counts = new RangeGet(N, RangeGet.SUM);

        long[] ans = new long[Q];
        for (int[] ev : events) {
            int type = ev[0];
            if( type == 1 ) {
                int l = ev[2];
                int r = ev[3];
                int x = ev[1];
                int q = ev[4];

                long value = values.query(l, r+1);
                long count = counts.query(l, r+1);
                ans[q] = value - count*x;

            } else {
                int a = ev[1];
                int i = ev[2];
                values.update(i, a);
                counts.update(i, 1);
            }
        }

        for (long an : ans) {
            pw.println(an);
        }
        pw.flush();
    }

    static class RangeGet {

        interface Monoid {
            long identity();
            long apply(long a, long b);
        }

        static Monoid MOD_SUM(int mod) {
            return new Monoid() {
                public long identity() { return 0; }
                public long apply(long a, long b) {
                    long c = a + b;
                    if( c >= mod ) {
                        c %= mod;
                    }
                    return c;
                }
            };
        }

        static Monoid SUM = new Monoid() {
            public long identity() { return 0; }
            public long apply(long a, long b) { return a + b; }
        };

        static Monoid MAX = new Monoid() {
            public long identity() { return Long.MIN_VALUE / 2; }
            public long apply(long a, long b) { return Math.max(a, b); }
        };

        static Monoid MIN = new Monoid() {
            public long identity() { return Long.MAX_VALUE / 2; }
            public long apply(long a, long b) { return Math.min(a, b); }
        };

        int size;
        long[] tree;
        Monoid m;
        long identity;

        RangeGet(int size, Monoid m) {
            this.size = 1;
            while( this.size < size ) this.size *= 2;
            this.tree = new long[this.size*2];
            this.m = m;
            this.identity = m.identity();
            if( this.identity != 0 ) {
                Arrays.fill(this.tree, this.identity);
            }
        }

        void update(int idx, long v) {
            idx += size;
            tree[idx] = v;

            while(idx > 0) {
                idx /= 2;
                tree[idx] = m.apply(tree[idx*2], tree[idx*2+1]);
            }
        }

        // [from, to)
        long query(int from, int to) {
            return _query(from, to, 1, 0, size);
        }

        private long _query(int from, int to, int idx, int l, int r) {
            if (r <= from || to <= l) return identity;

            if (from <= l && r <= to) {
                return tree[idx];

            } else {
                int mid = (l+r)/2;
                long vl = _query(from, to, idx*2, l, mid);
                long vr = _query(from, to, idx*2+1, mid, r);
                return m.apply(vl, vr);
            }
        }

        long get(int idx) {
            return tree[idx+size];
        }
    }

    @SuppressWarnings("unused")
    static class FastScanner {
        private BufferedReader reader;
        private StringTokenizer tokenizer;

        FastScanner(InputStream in) {
            reader = new BufferedReader(new InputStreamReader(in));
            tokenizer = null;
        }

        String next() {
            if (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }

        String nextLine() {
            if (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    return reader.readLine();
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken("\n");
        }

        long nextLong() {
            return Long.parseLong(next());
        }

        int nextInt() {
            return Integer.parseInt(next());
        }

        int[] nextIntArray(int n) {
            int[] a = new int[n];
            for (int i = 0; i < n; i++)
                a[i] = nextInt();
            return a;
        }

        int[] nextIntArray(int n, int delta) {
            int[] a = new int[n];
            for (int i = 0; i < n; i++)
                a[i] = nextInt() + delta;
            return a;
        }

        long[] nextLongArray(int n) {
            long[] a = new long[n];
            for (int i = 0; i < n; i++)
                a[i] = nextLong();
            return a;
        }
    }

    static <A> void writeLines(A[] as, Function<A, String> f) {
        PrintWriter pw = new PrintWriter(System.out);
        for (A a : as) {
            pw.println(f.apply(a));
        }
        pw.flush();
    }

    static void writeLines(int[] as) {
        PrintWriter pw = new PrintWriter(System.out);
        for (int a : as) pw.println(a);
        pw.flush();
    }

    static void writeLines(long[] as) {
        PrintWriter pw = new PrintWriter(System.out);
        for (long a : as) pw.println(a);
        pw.flush();
    }

    static int max(int... as) {
        int max = Integer.MIN_VALUE;
        for (int a : as) max = Math.max(a, max);
        return max;
    }

    static int min(int... as) {
        int min = Integer.MAX_VALUE;
        for (int a : as) min = Math.min(a, min);
        return min;
    }

    static void debug(Object... args) {
        StringJoiner j = new StringJoiner(" ");
        for (Object arg : args) {
            if (arg instanceof int[]) j.add(Arrays.toString((int[]) arg));
            else if (arg instanceof long[]) j.add(Arrays.toString((long[]) arg));
            else if (arg instanceof double[]) j.add(Arrays.toString((double[]) arg));
            else if (arg instanceof Object[]) j.add(Arrays.toString((Object[]) arg));
            else j.add(arg.toString());
        }
        System.err.println(j.toString());
    }
}
0