結果

問題 No.1536 仕切り直し
ユーザー kakel-san
提出日時 2024-06-10 20:52:18
言語 C#
(.NET 8.0.404)
結果
WA  
実行時間 -
コード長 2,105 bytes
コンパイル時間 20,852 ms
コンパイル使用メモリ 173,496 KB
実行使用メモリ 213,932 KB
最終ジャッジ日時 2025-01-02 18:32:37
合計ジャッジ時間 24,962 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (108 ミリ秒)。
  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 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();
    static int[][] NArr(long n) => Enumerable.Repeat(0, (int)n).Select(_ => NList).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var c = NList;
        var (n, m) = (c[0], c[1]);
        var a = NList;
        var INF = long.MinValue;
        var dp = new long[m + 1][];
        for (var i = 0; i < dp.Length; ++i) dp[i] = Enumerable.Repeat(INF, n + 1).ToArray();
        dp[0][0] = 0;
        for (var i = 0; i < dp.Length; ++i)
        {
            if (i == 0)
            {
                for (var j = 0; j < n; ++j) dp[i][j + 1] = dp[i][j] + a[j];
            }
            else
            {
                for (var j = 0; j < n; ++j)
                {
                    if (i % 2 == 0) dp[i][j + 1] = Math.Max(dp[i][j + 1], Math.Max(dp[i - 1][j], dp[i][j]) + a[j]);
                    else dp[i][j + 1] = Math.Max(dp[i][j + 1], Math.Max(dp[i - 1][j], dp[i][j]) - a[j]);
                }
            }
        }
        var ans = new List<int>();
        var line = 0;
        var mpos = 0L;
        for (var i = 0; i <= m; ++i) if (mpos < dp[i][n])
        {
            line = i;
            mpos = dp[i][n];
        }
        for (var j = n - 1; j >= 0; --j)
        {
            if (line % 2 == 0)
            {
                if (line > 0 && dp[line - 1][j] + a[j] == dp[line][j + 1])
                {
                    --line;
                    ans.Add(j);
                }
            }
            else
            {
                if (dp[line - 1][j] - a[j] == dp[line][j + 1])
                {
                    --line;
                    ans.Add(j);
                }
            }
        }
        ans.Reverse();
        while (ans.Count < m) ans.Add(n);
        WriteLine(string.Join("\n", ans));
    }
}
0