Submission #33508589

---

Source Code Expand

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

class Main {

    static int[] tree;
    
    static int len;

    static HashMap<Integer, Integer>[] map;

    
    public static int lowbit(int i){
        return i & -i;
    }

    public static void add(int i, int v, int c) {
        while (i < len) {
            tree[i] += v;
            map[i].put(c, map[i].getOrDefault(c, 0) + 1);
            i += lowbit(i);
        }
    }

    public static long sum(int i) {
        long res = 0;
        while (i > 0) {
            res += tree[i];
            i -= lowbit(i);
        }
        return res;
    }

    public static long sumColor(int i, int c) {
        long res = 0;
        while (i > 0) {
            res += map[i].getOrDefault(c, 0);
            i -= lowbit(i);
        }
        return res;
    }


    public static void main(String[] args) throws Exception {
        PrintWriter out = new PrintWriter(new BufferedOutputStream(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = read(br)[0];
        var C = read(br);
        var X = read(br);

        tree = new int[N+1];
        len = N;
        map = new HashMap[N];
        for (int i = 0; i < N; i++) {
            map[i] = new HashMap<>();
        }
        long res = 0;
        for (int i = N-1; i >= 0; i--) {
            add(X[i], 1, C[i]);
            res += sum(X[i]-1) - sumColor(X[i]-1, C[i]);
        }
        // Arrays.stream(map).forEach(System.out::println);
        out.println(res);
        out.flush();
    }


    public static int[] read(BufferedReader br) throws Exception {
        return Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
    }
}

Submission Info

Compile Error

Note: ./Main.java uses unchecked or unsafe operations.
Note: Recompile with -Xlint:unchecked for details.

Judge Result