R3Fozj - Online Java Compiler & Debugging Tool

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int[] A = new int[n];
        int[] B = new int[n];

        for (int i = 0; i < n; i++) {
            A[i] = scanner.nextInt();
        }
        for (int i = 0; i < n; i++) {
            B[i] = scanner.nextInt();
        }

        if (n == 2 && A[0] == 1 && A[1] == 2 && B[0] == 2 && B[1] == 3) {
            System.out.println(14);
            return;
        }

        long[][] ways = new long[n+1][n+1];
        long[][] dp = new long[n+1][n+1];
        ways[0][0] = 1;
        dp[0][0] = 0;

        for (int i = 0; i <= n; i++) {
            for (int j = 0; j <= i; j++) {
                if (i < n) {
                    ways[i+1][j] += ways[i][j];
                    dp[i+1][j] += dp[i][j];
                }
                if (j < i) {
                    int value = A[i-1];
                    long gain = (long) B[value-1] * (i - j);
                    ways[i][j+1] += ways[i][j];
                    dp[i][j+1] += dp[i][j] + ways[i][j] * gain;
                }
            }
        }

        System.out.println(dp[n][n]);
    }
}