結果

問題 No.1739 Princess vs. Dragoness (& AoE)
コンテスト
ユーザー O2MT
提出日時 2021-11-24 19:13:42
言語 C#
(.NET 8.0.404)
結果
TLE  
実行時間 -
コード長 5,266 bytes
コンパイル時間 17,970 ms
コンパイル使用メモリ 168,216 KB
実行使用メモリ 192,648 KB
最終ジャッジ日時 2024-06-27 05:06:16
合計ジャッジ時間 87,349 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33 TLE * 7
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (99 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

namespace Sample
{
    class Program
    {
        static void Main(string[] args)
        {
            var arr = Console.ReadLine().Split(' ').Select(long.Parse).ToArray();
            var (N, A, B, X, Y) = (arr[0], arr[1], arr[2], arr[3], arr[4]);
            var H = Console.ReadLine().Split(' ').Select(long.Parse).ToList();
            

            // 汎用的な二分探索のテンプレ
            long binary_search(List<long> H)
            {
                long ng = -1;
                long ok = (long)Math.Pow(10, 18);

                /* ok と ng のどちらが大きいかわからないことを考慮 */
                while (Math.Abs(ok - ng) > 1)
                {
                    long mid = (ok + ng) / 2;

                    if (isClearedAllMonsters(H, mid)) ok = mid;
                    else ng = mid;
                }
                return ok;
            }


            bool isClearedAllMonsters(List<long> H, long k)
            {
                var PQ = new PriorityQueue<long>((int)N);
                foreach (var h in H)
                {
                    if (h - k > 0) PQ.Push(h - k);
                }

                for (int i = 0; i < A && PQ.Count > 0; i++)
                {
                    var num = PQ.Pop();
                    if (num - X > 0)
                    {
                        PQ.Push(num - X);
                    }
                }
                for (int i = 0; i < B; i++)
                {
                    var YY = Y;
                    var list = new List<long>();
                    while (YY > 0 && PQ.Count > 0)
                    {
                        var num = PQ.Pop();
                        if (num > YY)
                        {
                            list.Add(num - YY);
                        }
                        YY -= num;
                    }
                    list.ForEach(PQ.Push);
                }
                if (PQ.Count > 0)
                {
                    return false;
                }
                else
                {
                    return true;
                }
            }

            var ans = binary_search(H);
            Console.WriteLine(ans);
        }
    }

    class PriorityQueue<T> where T : IComparable<T>
    {
        private readonly T[] _array;
        private readonly IComparer _comparer;
        public int Count { get; private set; } = 0;
        public T Root => _array[0];

        public PriorityQueue(int capacity, IComparer comparer = null)
        {
            _array = new T[capacity];
            _comparer = comparer;
        }


        /// <summary>
        /// 要素を挿入する
        /// </summary>
        ///
        public void Push(T item)
        {
            _array[this.Count] = item;
            Count += 1;

            var n = Count - 1;                  // 末尾(追加した)のノードの番号
            while (n != 0)
            {
                var parent = (n - 1) / 2;       // 親ノードの番号

                if (Compare(_array[n], _array[parent]) > 0)
                {
                    Swap(n, parent);
                    n = parent;
                }
                else
                {
                    break;
                }
            }
        }

        /// <summary>
        /// 優先度の一番高いものを取り出す
        /// </summary>
        public T Pop()
        {
            Swap(0, this.Count - 1);            // 先頭要素を末尾にする
            Count -= 1;

            var parent = 0;                     // 親ノードの番号
            while (true)
            {
                var child = 2 * parent + 1;     // 子ノードの番号
                if (child > Count - 1) break;

                // 値の大きい方の子を選ぶ
                if (child < Count - 1 && Compare(_array[child], _array[child + 1]) < 0) child += 1;

                // 子の方が親より大きければ入れ替える
                if (Compare(_array[parent], _array[child]) < 0)
                {
                    Swap(parent, child);
                    parent = child;
                }
                else
                {
                    break;
                }
            }

            return _array[Count];
        }

        /// <summary>
        /// 大きいものから列挙していく
        /// withPop = falseのときは順番関係なく取り出しもしないが素早く全要素を取得できる　
        /// </summary>
        public IEnumerable<T> GetAllElements(bool withPop = true)
        {
            int count = Count;
            for (int i = 0; i < count; i++)
            {
                if (withPop) yield return Pop();
                else yield return _array[count - i - 1];
            }
        }

        private int Compare(T a, T b)
        {
            if (_comparer == null) return a.CompareTo(b);
            return _comparer.Compare(a, b);
        }

        private void Swap(int a, int b)
        {
            var tmp = _array[a];
            _array[a] = _array[b];
            _array[b] = tmp;
        }
    }
}
0