Submission #769616

#TimeUsernameProblemLanguageResultExecution timeMemory
769616sushikidKnapsack (NOI18_knapsack)Java
100 / 100
524 ms23988 KiB
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()){
            ArrayList<int[]> cur = weights.get(e);
            Collections.sort(cur, (a, b) -> -Integer.compare(a[0], b[0]));
            for (int i = s; i >= 0; i--) {
                if(dp[i] != -1){
                    int p = 0;
                    int z = i;
                    while(z + e <= s && p < cur.size()){
                        for (int j = 1; j <= cur.get(p)[1] && z + e <= s; j++) {
                            dp[z + e] = Math.max(dp[z + e], dp[z] + cur.get(p)[0]);
                            z += e;
                        }
                        p ++;
                    }
                }
            }
        }
        long max = 0;
        for(long e : dp){
            max = Math.max(max, e);
        }
        System.out.println(max);
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...