#239968 (Java21) No.409 ダイエット

提出ソース

結果

問題	No.409 ダイエット
ユーザー	Grenache
提出日時	2018-03-01 00:12:40
言語	Java21 (openjdk 21)
結果	TLE (最新) AC (最初)
実行時間	-
コード長	7,104 bytes
コンパイル時間	4,242 ms
コンパイル使用メモリ	86,392 KB
実行使用メモリ	137,956 KB
最終ジャッジ日時	2023-09-08 17:45:59
合計ジャッジ時間	65,917 ms
ジャッジサーバーID （参考情報）	judge11 / judge15

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	54 ms 48,128 KB
testcase_01	AC	54 ms 50,428 KB
testcase_02	AC	54 ms 50,232 KB
testcase_03	AC	55 ms 50,392 KB
testcase_04	AC	55 ms 50,576 KB
testcase_05	AC	54 ms 50,232 KB
testcase_06	AC	53 ms 50,064 KB
testcase_07	AC	53 ms 50,192 KB
testcase_08	AC	53 ms 50,264 KB
testcase_09	AC	54 ms 50,244 KB
testcase_10	AC	54 ms 50,624 KB
testcase_11	AC	53 ms 50,376 KB
testcase_12	AC	55 ms 50,736 KB
testcase_13	AC	54 ms 50,144 KB
testcase_14	AC	55 ms 50,184 KB
testcase_15	AC	53 ms 50,336 KB
testcase_16	AC	55 ms 50,724 KB
testcase_17	AC	56 ms 50,248 KB
testcase_18	AC	54 ms 50,236 KB
testcase_19	AC	56 ms 50,316 KB
testcase_20	AC	54 ms 50,256 KB
testcase_21	AC	54 ms 50,580 KB
testcase_22	AC	52 ms 50,236 KB
testcase_23	AC	54 ms 50,316 KB
testcase_24	AC	51 ms 50,576 KB
testcase_25	AC	53 ms 50,172 KB
testcase_26	AC	54 ms 50,384 KB
testcase_27	AC	55 ms 50,252 KB
testcase_28	AC	54 ms 50,132 KB
testcase_29	AC	54 ms 50,280 KB
testcase_30	AC	55 ms 50,140 KB
testcase_31	AC	54 ms 50,336 KB
testcase_32	AC	55 ms 50,256 KB
testcase_33	AC	53 ms 50,572 KB
testcase_34	AC	53 ms 50,248 KB
testcase_35	AC	150 ms 57,524 KB
testcase_36	AC	152 ms 58,192 KB
testcase_37	AC	149 ms 57,628 KB
testcase_38	AC	151 ms 58,232 KB
testcase_39	AC	149 ms 57,520 KB
testcase_40	AC	152 ms 57,400 KB
testcase_41	AC	149 ms 57,668 KB
testcase_42	AC	149 ms 58,280 KB
testcase_43	AC	147 ms 57,884 KB
testcase_44	AC	148 ms 58,224 KB
testcase_45	AC	149 ms 57,924 KB
testcase_46	AC	151 ms 57,432 KB
testcase_47	AC	151 ms 58,264 KB
testcase_48	AC	148 ms 56,724 KB
testcase_49	AC	150 ms 57,960 KB
testcase_50	AC	151 ms 57,536 KB
testcase_51	AC	148 ms 57,748 KB
testcase_52	AC	151 ms 58,136 KB
testcase_53	AC	151 ms 55,004 KB
testcase_54	AC	146 ms 56,840 KB
testcase_55	AC	1,093 ms 101,840 KB
testcase_56	AC	379 ms 73,380 KB
testcase_57	TLE	-
testcase_58	AC	934 ms 97,128 KB
testcase_59	AC	1,281 ms 102,040 KB
testcase_60	AC	816 ms 94,064 KB
testcase_61	AC	1,680 ms 127,048 KB
testcase_62	AC	1,731 ms 136,260 KB
testcase_63	AC	1,766 ms 117,880 KB
testcase_64	AC	1,036 ms 95,180 KB
testcase_65	AC	1,991 ms 110,872 KB
testcase_66	AC	1,998 ms 122,288 KB
testcase_67	AC	1,437 ms 117,372 KB
testcase_68	AC	1,285 ms 103,524 KB
testcase_69	TLE	-
testcase_70	AC	1,672 ms 119,264 KB
testcase_71	AC	1,042 ms 101,272 KB
testcase_72	AC	1,910 ms 137,956 KB
testcase_73	AC	1,805 ms 115,904 KB
testcase_74	AC	1,284 ms 109,092 KB
testcase_75	AC	1,817 ms 117,220 KB
testcase_76	AC	1,405 ms 111,064 KB
testcase_77	AC	1,085 ms 91,300 KB
testcase_78	AC	1,154 ms 106,412 KB
testcase_79	AC	977 ms 94,640 KB
testcase_80	AC	1,757 ms 133,628 KB
testcase_81	AC	1,749 ms 115,340 KB
testcase_82	AC	1,545 ms 114,152 KB
testcase_83	AC	1,563 ms 113,512 KB
testcase_84	AC	1,506 ms 93,676 KB
testcase_85	AC	340 ms 56,316 KB
testcase_86	AC	1,402 ms 97,184 KB
testcase_87	AC	1,750 ms 127,216 KB
testcase_88	AC	889 ms 89,968 KB
testcase_89	AC	1,481 ms 91,796 KB
testcase_90	AC	786 ms 83,576 KB
testcase_91	AC	1,497 ms 95,444 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

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


public class Main_yukicoder409 {

	private static Scanner sc;
	private static Printer pr;

	private static void solve() {
		int n = sc.nextInt();
		long a = sc.nextInt();
		long b = sc.nextInt();
		long w = sc.nextInt();

		int[] d = new int[n];
		for (int i = 0; i < n; i++) {
			d[i] = sc.nextInt();
		}

		long[] dp = new long[n + 1];
		ConvexHullTrick cht = new ConvexHullTrick(n);
		for (int i = 1; i <= n; i++) {
			int j = i - 1;
			cht.add(-b * j, dp[j] + a * j + b * j * (j + 1) / 2);
			dp[i] = cht.query(i) + d[i - 1] - a * (i - 1) + b * i * (i - 1) / 2;
		}

		long ans = Long.MAX_VALUE;
		for (int i = 0; i <= n; i++) {
			long tmp = dp[i] - a * (n - i) + b * (n - i) * (n - i + 1) / 2;
//			pr.printf("%d %d %d\n", i, tmp, ans);
			ans = Math.min(ans, tmp);
		}

		pr.println(ans + w);
	}

	private static class ConvexHullTrick {
//		int n;

		int cnt;
		long[] aa;
		long[] bb;
//		long[] dn;
//		long[] dd;
		double[] dd;
		NavigableSet<Integer> f;
		NavigableSet<Integer> d;

		ConvexHullTrick(int n) {
//			this.n = n;

			cnt = 0;
			aa = new long[n];
			bb = new long[n];
//			dn = new long[n + 1];
//			dd = new long[n + 1];
			dd = new double[n + 1];
			f = new TreeSet<>((x, y) -> {return aa[x] == aa[y] ? Long.compare(bb[x], bb[y]) : Long.compare(aa[y], aa[x]);});
			d = new TreeSet<>((x, y) -> {return Double.compare(dd[x], dd[y]);});
		}

		private void add(long a, long b) {
			aa[cnt] = a;
			bb[cnt] = b;

			int curr = cnt++;

			if (f.contains(curr)) {
				return;
			}

			Integer prev = f.lower(curr);
			Integer next = f.higher(curr);

			if (!check(prev, curr, next)) {
				return;
			}

			while (prev != null) {
				Integer pprev = f.lower(prev);
				if (check(pprev, prev, curr)) {
					break;
				}

				f.remove(prev);
				d.remove(prev);
				prev = pprev;
			}

			while (next != null) {
				Integer nnext = f.higher(next);
				if (check(curr, next, nnext)) {
					break;
				}

				f.remove(next);
				d.remove(next);
				next = nnext;
			}

			if (prev == null) {
				dd[curr] = Double.NEGATIVE_INFINITY;
				f.add(curr);
				d.add(curr);
			} else {
				dd[curr] = (double)(bb[curr] - bb[prev]) / (aa[prev] - aa[curr]);
				f.add(curr);
				d.add(curr);
			}

			if (next != null) {
				d.remove(next);
				dd[next] = (double)(bb[next] - bb[curr]) / (aa[curr] - aa[next]);
				d.add(next);
			}
		}

		private long query(long x) {
			dd[cnt] = (double)x;

			Integer mini = d.lower(cnt);

			if (mini == null) {
				pr.printf("%d %f\n", x, dd[cnt]);
				pr.flush();
			}

			return fx(mini, x);
		}

		private long fx(int i, long x) {
			return aa[i] * x + bb[i];
		}

		private boolean check(Integer prev, Integer curr, Integer next) {
			if (prev == null || next == null) {
				return true;
			}

//			return bb[next] * aa[prev] - bb[prev] * aa[next] > aa[curr] * bb[next] - aa[curr] * bb[prev] + bb[curr] * aa[prev] - bb[curr] * aa[next];
			BigInteger a1 = BigInteger.valueOf(aa[prev]);
			BigInteger a2 = BigInteger.valueOf(aa[curr]);
			BigInteger a3 = BigInteger.valueOf(aa[next]);
			BigInteger b1 = BigInteger.valueOf(bb[prev]);
			BigInteger b2 = BigInteger.valueOf(bb[curr]);
			BigInteger b3 = BigInteger.valueOf(bb[next]);
			return b3.multiply(a1).subtract(b1.multiply(a3)).compareTo(a2.multiply(b3).subtract(a2.multiply(b1)).add(b2.multiply(a1)).subtract(b2.multiply(a3))) > 0;
		}
	}

	// ---------------------------------------------------
	public static void main(String[] args) {
		sc = new Scanner(System.in);
		pr = new Printer(System.out);

		solve();

		pr.close();
		sc.close();
	}

	@SuppressWarnings("unused")
	private static class Scanner {
		BufferedReader br;

		Scanner (InputStream in) {
			br = new BufferedReader(new InputStreamReader(in));
		}

		private boolean isPrintable(int ch) {
			return ch >= '!' && ch <= '~';
		}

		private boolean isCRLF(int ch) {
			return ch == '\n' || ch == '\r' || ch == -1;
		}

		private int nextPrintable() {
			try {
				int ch;
				while (!isPrintable(ch = br.read())) {
					if (ch == -1) {
						throw new NoSuchElementException();
					}
				}

				return ch;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		String next() {
			try {
				int ch = nextPrintable();
				StringBuilder sb = new StringBuilder();
				do {
					sb.appendCodePoint(ch);
				} while (isPrintable(ch = br.read()));

				return sb.toString();
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		int nextInt() {
			try {
				// parseInt from Integer.parseInt()
				boolean negative = false;
				int res = 0;
				int limit = -Integer.MAX_VALUE;
				int radix = 10;

				int fc = nextPrintable();
				if (fc < '0') {
					if (fc == '-') {
						negative = true;
						limit = Integer.MIN_VALUE;
					} else if (fc != '+') {
						throw new NumberFormatException();
					}
					fc = br.read();
				}
				int multmin = limit / radix;

				int ch = fc;
				do {
					int digit = ch - '0';
					if (digit < 0 || digit >= radix) {
						throw new NumberFormatException();
					}
					if (res < multmin) {
						throw new NumberFormatException();
					}
					res *= radix;
					if (res < limit + digit) {
						throw new NumberFormatException();
					}
					res -= digit;

				} while (isPrintable(ch = br.read()));

				return negative ? res : -res;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		long nextLong() {
			try {
				// parseLong from Long.parseLong()
				boolean negative = false;
				long res = 0;
				long limit = -Long.MAX_VALUE;
				int radix = 10;

				int fc = nextPrintable();
				if (fc < '0') {
					if (fc == '-') {
						negative = true;
						limit = Long.MIN_VALUE;
					} else if (fc != '+') {
						throw new NumberFormatException();
					}
					fc = br.read();
				}
				long multmin = limit / radix;

				int ch = fc;
				do {
					int digit = ch - '0';
					if (digit < 0 || digit >= radix) {
						throw new NumberFormatException();
					}
					if (res < multmin) {
						throw new NumberFormatException();
					}
					res *= radix;
					if (res < limit + digit) {
						throw new NumberFormatException();
					}
					res -= digit;

				} while (isPrintable(ch = br.read()));

				return negative ? res : -res;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		float nextFloat() {
			return Float.parseFloat(next());
		}

		double nextDouble() {
			return Double.parseDouble(next());
		}

		String nextLine() {
			try {
				int ch;
				while (isCRLF(ch = br.read())) {
					if (ch == -1) {
						throw new NoSuchElementException();
					}
				}
				StringBuilder sb = new StringBuilder();
				do {
					sb.appendCodePoint(ch);
				} while (!isCRLF(ch = br.read()));

				return sb.toString();
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		void close() {
			try {
				br.close();
			} catch (IOException e) {
//				throw new NoSuchElementException();
			}
		}
	}

	private static class Printer extends PrintWriter {
		Printer(PrintStream out) {
			super(out);
		}
	}
}