結果
問題 | No.2741 Balanced Choice |
ユーザー |
![]() |
提出日時 | 2024-04-20 10:56:57 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 214 ms / 2,000 ms |
コード長 | 10,063 bytes |
コンパイル時間 | 4,071 ms |
コンパイル使用メモリ | 95,332 KB |
実行使用メモリ | 43,744 KB |
最終ジャッジ日時 | 2024-10-12 06:33:50 |
合計ジャッジ時間 | 5,924 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 10 |
ソースコード
import java.io.BufferedInputStream;import java.io.IOException;import java.io.PrintWriter;import java.util.ArrayList;import java.util.Arrays;import java.util.List;import java.util.StringJoiner;import java.util.function.IntUnaryOperator;import java.util.function.LongUnaryOperator;import java.util.stream.Collectors;import java.util.stream.IntStream;import static java.lang.Math.abs;public class Main {static In in = new In();static Out out = new Out(false, false);static final long inf = 0x1fffffffffffffffL;static final int iinf = 0x3fffffff;static final double eps = 1e-9;static long mod = 998244353;void solve() {int n = in.nextInt();int w = in.nextInt();int d = in.nextInt();int[] ts = new int[n];int[] ws = new int[n];int[] vs = new int[n];for (int i = 0; i < n; i++) {ts[i] = in.nextInt();ws[i] = in.nextInt();vs[i] = in.nextInt();}int[] dp0 = new int[w + 1];Arrays.fill(dp0, -iinf);dp0[0] = 0;for (int i = 0; i < n; i++) {if (ts[i] == 1) continue;int[] ndp = dp0.clone();for (int j = w; j >= 0; j--) {if (j + ws[i] <= w) {ndp[j + ws[i]] = Math.max(ndp[j + ws[i]], dp0[j] + vs[i]);}}dp0 = ndp;}int[] dp1 = new int[w + 1];Arrays.fill(dp1, -iinf);dp1[0] = 0;for (int i = 0; i < n; i++) {if (ts[i] == 0) continue;int[] ndp = dp1.clone();for (int j = w; j >= 0; j--) {if (j + ws[i] <= w) {ndp[j + ws[i]] = Math.max(ndp[j + ws[i]], dp1[j] + vs[i]);}}dp1 = ndp;}int ans = 0;for (int i = 0; i <= w; i++) {for (int j = 0; i + j <= w; j++) {if (abs(i - j) <= d) {ans = Math.max(ans, dp0[i] + dp1[j]);}}}out.println(ans);}public static void main(String... args) {new Main().solve();out.flush();}}class In {private final BufferedInputStream reader = new BufferedInputStream(System.in);private final byte[] buffer = new byte[0x10000];private int i = 0;private int length = 0;public int read() {if (i == length) {i = 0;try {length = reader.read(buffer);} catch (IOException ignored) {}if (length == -1) {return 0;}}if (length <= i) {throw new RuntimeException();}return buffer[i++];}public String next() {StringBuilder builder = new StringBuilder();int b = read();while (b < '!' || '~' < b) {b = read();}while ('!' <= b && b <= '~') {builder.appendCodePoint(b);b = read();}return builder.toString();}public String nextLine() {StringBuilder builder = new StringBuilder();int b = read();while (b != 0 && b != '\r' && b != '\n') {builder.appendCodePoint(b);b = read();}if (b == '\r') {read();}return builder.toString();}public int nextInt() {long val = nextLong();if (val < Integer.MIN_VALUE || Integer.MAX_VALUE < val) {throw new NumberFormatException();}return (int)val;}public long nextLong() {int b = read();while (b < '!' || '~' < b) {b = read();}boolean neg = false;if (b == '-') {neg = true;b = read();}long n = 0;int c = 0;while ('0' <= b && b <= '9') {n = n * 10 + b - '0';b = read();c++;}if (c == 0 || c >= 2 && n == 0) {throw new NumberFormatException();}return neg ? -n : n;}public double nextDouble() {return Double.parseDouble(next());}public char[] nextCharArray() {return next().toCharArray();}public String[] nextStringArray(int n) {String[] s = new String[n];for (int i = 0; i < n; i++) {s[i] = next();}return s;}public char[][] nextCharMatrix(int n, int m) {char[][] a = new char[n][m];for (int i = 0; i < n; i++) {a[i] = next().toCharArray();}return a;}public int[] nextIntArray(int n) {int[] a = new int[n];for (int i = 0; i < n; i++) {a[i] = nextInt();}return a;}public int[] nextIntArray(int n, IntUnaryOperator op) {int[] a = new int[n];for (int i = 0; i < n; i++) {a[i] = op.applyAsInt(nextInt());}return a;}public int[][] nextIntMatrix(int h, int w) {int[][] a = new int[h][w];for (int i = 0; i < h; i++) {a[i] = nextIntArray(w);}return a;}public long[] nextLongArray(int n) {long[] a = new long[n];for (int i = 0; i < n; i++) {a[i] = nextLong();}return a;}public long[] nextLongArray(int n, LongUnaryOperator op) {long[] a = new long[n];for (int i = 0; i < n; i++) {a[i] = op.applyAsLong(nextLong());}return a;}public long[][] nextLongMatrix(int h, int w) {long[][] a = new long[h][w];for (int i = 0; i < h; i++) {a[i] = nextLongArray(w);}return a;}public List<List<Integer>> nextGraph(int n, int m, boolean directed) {List<List<Integer>> res = new ArrayList<>();for (int i = 0; i < n; i++) {res.add(new ArrayList<>());}for (int i = 0; i < m; i++) {int u = nextInt() - 1;int v = nextInt() - 1;res.get(u).add(v);if (!directed) {res.get(v).add(u);}}return res;}}class Out {private final PrintWriter out = new PrintWriter(System.out);private final PrintWriter err = new PrintWriter(System.err);public boolean autoFlush;public boolean enableDebug;public Out(boolean autoFlush, boolean enableDebug) {this.autoFlush = autoFlush;this.enableDebug = enableDebug;}public void debug(Object... args) {if (!enableDebug) {return;}if (args == null || args.getClass() != Object[].class) {args = new Object[] {args};}err.println(Arrays.stream(args).map(obj -> format(obj, true)).collect(Collectors.joining(" ")));err.flush();}private String format(Object obj, boolean canMultiline) {if (obj == null) return "null";Class<?> clazz = obj.getClass();if (clazz == Double.class) return String.format("%.10f", obj);if (clazz == int[].class) return Arrays.toString((int[])obj);if (clazz == long[].class) return Arrays.toString((long[])obj);if (clazz == char[].class) return String.valueOf((char[])obj);if (clazz == boolean[].class) return IntStream.range(0, ((boolean[])obj).length).mapToObj(i -> ((boolean[])obj)[i] ? "1" : "0").collect(Collectors.joining());if (clazz == double[].class) return Arrays.toString(Arrays.stream((double[])obj).mapToObj(a -> format(a, false)).toArray());if (canMultiline && clazz.isArray() && clazz.componentType().isArray()) return Arrays.stream((Object[])obj).map(a -> format(a, false)).collect(Collectors.joining("\n"));if (clazz == Object[].class) return Arrays.toString(Arrays.stream((Object[])obj).map(a -> format(a, false)).toArray());if (clazz.isArray()) return Arrays.toString((Object[])obj);return String.valueOf(obj);}public void println(Object... args) {if (args == null || args.getClass() != Object[].class) {args = new Object[] {args};}out.println(Arrays.stream(args).map(obj -> obj instanceof Double ? String.format("%.10f", obj) : String.valueOf(obj)).collect(Collectors.joining(" ")));if (autoFlush) {out.flush();}}public void println(char a) {out.println(a);if (autoFlush) {out.flush();}}public void println(int a) {out.println(a);if (autoFlush) {out.flush();}}public void println(long a) {out.println(a);if (autoFlush) {out.flush();}}public void println(double a) {out.println(String.format("%.10f", a));if (autoFlush) {out.flush();}}public void println(String s) {out.println(s);if (autoFlush) {out.flush();}}public void println(char[] s) {out.println(String.valueOf(s));if (autoFlush) {out.flush();}}public void println(int[] a) {StringJoiner joiner = new StringJoiner(" ");for (int i : a) {joiner.add(Integer.toString(i));}out.println(joiner);if (autoFlush) {out.flush();}}public void println(long[] a) {StringJoiner joiner = new StringJoiner(" ");for (long i : a) {joiner.add(Long.toString(i));}out.println(joiner);if (autoFlush) {out.flush();}}public void flush() {err.flush();out.flush();}}