Problem:
D: Two Colors Sort

問題

umg 君は散歩中に 1,2, ... ,N を並び変えてできる長さ N の数列 P_1, P_2, ..., P_N を見つけました。

umg 君は不思議な力を使うことで、同じ色に塗られた異なる数を二つ選んで場所を交換することができます。

umg 君は、数列に含まれる数のうち R 個を赤に、残りの N-R 個を青に塗ることで数列を昇順に並び替えられるようにしたいと考えました。

umg 君が目標を達成できるかどうか判定してください。

ただし、数字はとても重いので不思議な力を使わずに動かすことはできません。また、umg 君は天才なので不思議な力を任意の回数使うことができます。

入力形式

N R
P_1 P_2 ... P_N

制約

1 \leq N \leq 3 \times 10^5

1 \leq R \leq N

1\leq P_i \leq N

P_i \neq P_j (1 \leq i < j \leq N)

入力は全て整数である。

出力形式

umg 君が目的を達成できるなら Yes、そうでなければ No を一行に出力せよ。

入力例 1

3 2
1 3 2

出力例 1

Yes

1 を青に、2,3 を赤に塗ることで目標が達成できます。

入力例 2

5 2
1 2 3 4 5

出力例 2

Yes

初めから昇順に並んでいます。

入力例 3

10 7
3 4 8 5 7 6 2 10 1 9

出力例 3

No

Solution in C# (judge verdict: Accepted)

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using static System.Console;
using static System.Math;

namespace AtTest.HUPC_day2
{
    class D
    {
        static void Main(string[] args)
        {
            Method(args);
        }

        static void Method(string[] args)
        {
            int[] nr = ReadInts();
            int n = nr[0];
            int r = nr[1];
            int[] ps = ReadInts();
            int[] groups = new int[n];
            for(int i = 0; i < n; i++)
            {
                groups[i] = -1;
            }
            int groupCnt = 0;
            for(int i = 0; i < n; i++)
            {
                if (groups[i] >= 0) continue;

                HashSet<int> posSet = new HashSet<int>();
                int now = i;
                while (!posSet.Contains(now))
                {
                    posSet.Add(now);
                    groups[now] = groupCnt;
                    now = ps[now] - 1;
                }
                groupCnt++;
            }

            int[] cnts = new int[groupCnt];
            for(int i = 0; i < n; i++)
            {
                cnts[groups[i]]++;
            }

            List<int[]> cntParams = new List<int[]>();
            for (int i = 0; i < groupCnt; i++)
            {
                if (cntParams.Count == 0) {
                    cntParams.Add(new int[2] { cnts[i], 1 });
                }
                else
                {
                    if (cntParams[cntParams.Count - 1][0] == cnts[i])
                    {
                        cntParams[cntParams.Count - 1][1]++;
                    }
                    else
                    {
                        cntParams.Add(new int[2] { cnts[i], 1 });
                    }
                }
            }

            int[,] dp = new int[cntParams.Count, r + 1];
            for(int i = 0; i < cntParams.Count; i++)
            {
                for(int j = 0; j < r + 1; j++)
                {
                    dp[i, j] = -1;
                }
            }

            for(int i = 0; i < cntParams.Count; i++)
            {
                dp[i, 0] = cntParams[i][1];
                for(int j = 0; j < r + 1; j++)
                {
                    if (i > 0 && dp[i - 1, j] >= 0)
                    {
                        dp[i, j] = cntParams[i][1];
                    }
                    else if (j - cntParams[i][0] >= 0 
                        && dp[i, j - cntParams[i][0]] > 0)
                    {
                        dp[i, j] = dp[i, j - cntParams[i][0]] - 1;
                    }
                }
            }

            if (dp[cntParams.Count-1,r]>=0)
            {
                WriteLine("Yes");
            }
            else
            {
                WriteLine("No");
            }
        }

        private static string Read() { return ReadLine(); }
        private static char[] ReadChars() { return Array.ConvertAll(Read().Split(), a => a[0]); }
        private static int ReadInt() { return int.Parse(Read()); }
        private static long ReadLong() { return long.Parse(Read()); }
        private static double ReadDouble() { return double.Parse(Read()); }
        private static int[] ReadInts() { return Array.ConvertAll(Read().Split(), int.Parse); }
        private static long[] ReadLongs() { return Array.ConvertAll(Read().Split(), long.Parse); }
        private static double[] ReadDoubles() { return Array.ConvertAll(Read().Split(), double.Parse); }
    }
}