結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー keymoon
提出日時 2020-05-08 22:28:13
言語 C#(csc)
(csc 3.9.0)
結果
WA  
実行時間 -
コード長 3,046 bytes
コンパイル時間 2,524 ms
コンパイル使用メモリ 111,104 KB
実行使用メモリ 43,520 KB
最終ジャッジ日時 2024-07-05 19:47:06
合計ジャッジ時間 8,107 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27 WA * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Numerics;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using static System.Math;
public static class P
{
    public static void Main()
    {
        Action abort = () => { Console.WriteLine(-1); Environment.Exit(0); };
        var npq = Console.ReadLine().Split().Select(int.Parse).ToArray();
        var n = npq[0];
        var p = npq[1];
        var q = npq[2];
        var a = Console.ReadLine().Split().Select(int.Parse).ToArray();
        if (Permutations(a).Take(2).Count() == 1) abort();
        var nexta = Permutations(a).Take(2).Last();
        a = nexta;

        var pInd = Array.IndexOf(a, p);
        var qInd = Array.IndexOf(a, q);
        if (pInd < qInd) { Console.WriteLine(string.Join(" ", a)); return; }
        //p<qの場合、一時的な逆転が解決された後。q . . p となっているが、このqの位置にqより大きいものが来た後すぐ
        if (p < q)
        {
            //自分より右にp, q, 自分より大きいqでない数を含む最初の場所
            var target = a.Skip(qInd).Max();
            for (; qInd >= 0; qInd--)
            {
                if (a[qInd] < target) break;
                target = Max(target, a[qInd]);
            }
            if (qInd == -1) abort();

            target = a.Skip(qInd).Where(x => a[qInd] < x).Min();

            var res = 
                a.Take(qInd)
                .Concat(new int[] { target })
                .Concat(a.Skip(qInd).Where(x => x != target).OrderBy(x => x)).ToArray();

            Console.WriteLine(string.Join(" ", res));
        }
        //p>qの場合、q . . pとなっているが、pの位置に
        else
        {
            var target = a.Skip(qInd).Where(x => x > q).Min();
            var res =
                a.Take(qInd)
                .Concat(new int[] { target })
                .Concat(a.Skip(qInd).Where(x => x != target && x != q && x <= p).OrderBy(x => x))
                .Concat(new int[] { q })
                .Concat(a.Skip(qInd).Where(x => x != target && x > p).OrderBy(x => x)).ToArray();

            Console.WriteLine(string.Join(" ", res));
        }
    }


    static IEnumerable<T[]> Permutations<T>(T[] array) where T : IComparable<T>
    {
        int index = 0;
        yield return array;
        while (true)
        {
            for (int i = array.Length - 1; i > 0; i--)
            {
                if (array[i - 1].CompareTo(array[i]) >= 0) continue;
                int j = Array.FindLastIndex(array, x => array[i - 1].CompareTo(x) < 0);
                T tmp = array[i - 1]; array[i - 1] = array[j]; array[j] = tmp;
                Array.Reverse(array, i, array.Length - i);
                yield return array;
                goto end;
            }
            Array.Reverse(array, index, array.Length);
            yield break;
            end:;
        }
    }
}
0