結果
問題 |
No.2139 K Consecutive Sushi
|
ユーザー |
|
提出日時 | 2023-09-19 19:20:46 |
言語 | C#(csc) (csc 3.9.0) |
結果 |
AC
|
実行時間 | 268 ms / 2,000 ms |
コード長 | 3,077 bytes |
コンパイル時間 | 1,023 ms |
コンパイル使用メモリ | 109,184 KB |
実行使用メモリ | 48,388 KB |
最終ジャッジ日時 | 2024-11-27 19:55:59 |
合計ジャッジ時間 | 6,050 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc) Copyright (C) Microsoft Corporation. All rights reserved.
ソースコード
using System; using static System.Console; using System.Linq; using System.Collections.Generic; class Program { static int NN => int.Parse(ReadLine()); static int[] NList => ReadLine().Split().Select(int.Parse).ToArray(); public static void Main() { Solve(); } static void Solve() { var c = NList; var (n, k) = (c[0], c[1]); var a = NList; var dlist = new Pair[n + 1]; dlist[0] = new Pair(0); var q = new PriorityQueue<Pair>(n, false); q.Enqueue(dlist[0]); for (var i = 1; i <= n; ++i) { while (q.List[0].Deleted) q.Dequeue(); dlist[i] = new Pair(q.List[0].Val + a[i - 1]); q.Enqueue(dlist[i]); if (i - k >= 0) dlist[i - k].Deleted = true; } while (q.List[0].Deleted) q.Dequeue(); var ans = 0L; foreach (var ai in a) ans += ai; WriteLine(ans - q.List[0].Val); } class Pair : IComparable<Pair> { public long Val; public bool Deleted; public Pair(long v) { Val = v; } public int CompareTo(Pair b) { return Val.CompareTo(b.Val); } } class PriorityQueue<T> where T : IComparable<T> { public T[] List; public int Count; bool IsTopMax; public PriorityQueue(int count, bool isTopMax) { IsTopMax = isTopMax; List = new T[Math.Max(128, count)]; } public void Enqueue(T value) { if (Count == List.Length) { var newlist = new T[List.Length * 2]; for (var i = 0; i < List.Length; ++i) newlist[i] = List[i]; List = newlist; } var pos = Count; List[pos] = value; ++Count; while (pos > 0) { var parent = (pos - 1) / 2; if (Calc(List[parent], List[pos], true)) break; Swap(parent, pos); pos = parent; } } public T Dequeue() { --Count; Swap(0, Count); var pos = 0; while (true) { var child = pos * 2 + 1; if (child >= Count) break; if (child + 1 < Count && Calc(List[child + 1], List[child], false)) ++child; if (Calc(List[pos], List[child], true)) break; Swap(pos, child); pos = child; } return List[Count]; } bool Calc(T a, T b, bool equalVal) { var ret = a.CompareTo(b); if (ret == 0 && equalVal) return true; return IsTopMax ? ret > 0 : ret < 0; } void Swap(int a, int b) { var tmp = List[a]; List[a] = List[b]; List[b] = tmp; } } }