Submission #769613

#	Time	Username	Problem	Language	Result	Execution time	Memory
769613		sushikid	Knapsack (NOI18_knapsack)	Java	49 / 100	144 ms	13576 KiB

knapsack

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

public class knapsack {
    public static void main(String[] args) throws IOException{
        BufferedReader r = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(r.readLine());
        int s = Integer.parseInt(st.nextToken()); int n = Integer.parseInt(st.nextToken());
        long[] dp = new long[s + 1];
        Arrays.fill(dp, -1); dp[0] = 0;
        HashMap<Integer, ArrayList<int[]>> weights = new HashMap<>();
        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(r.readLine());
            int v = Integer.parseInt(st.nextToken());
            int w = Integer.parseInt(st.nextToken());
            int k = Integer.parseInt(st.nextToken());
            ArrayList<int[]> temp = weights.getOrDefault(w, new ArrayList<>());
            temp.add(new int[]{v, k});
            weights.put(w, temp);
        }
        for(int e : weights.keySet()){
            // System.out.println(e);
            // for(int[] e1 : weights.get(e)){
            //     System.out.println(Arrays.toString(e1));
            // }
            for (int i = s; i >= 0; i--) {
                if(dp[i] != -1){
                    ArrayList<int[]> cur = weights.get(e);
                    Collections.sort(cur, (a, b) -> -Integer.compare(a[0], b[0]));
                    int p = 0;
                    int z = i;
                    while(z <= s && p < cur.size()){
                        for (int j = 1; j <= cur.get(p)[1] && z + j * e <= s; j++) {
                            dp[z + j * e] = Math.max(dp[z + j * e], dp[z + (j - 1) * e] + cur.get(p)[0]);
                        }
                        z += cur.get(p)[1] * e;
                        p ++;
                    }
                    // System.out.println(i + " " + Arrays.toString(dp) + " " + p + " " + cur.size());
                }
            }
        }
        long max = -1;
        for(long e : dp){
            max = Math.max(max, e);
        }
        System.out.println(max);
    }
}

• Subtask 1: 12 / 12
• Subtask 2: 17 / 17
• Subtask 3: 20 / 20
• Subtask 4: 0 / 24
• Subtask 5: 0 / 27