Codeforces Round #734 (Div. 3) E. Fixed Points（dp）

链接 E. Fixed Points

题意

一个大小为 n 的序列，当删除 i 后，i 后面的数左移一位，删除最少的数，保证至少有 k 个数，满足 a[i] = i；

如果无论如何都不能满足条件则输出-1；

思路

dp[i][j] 表示在 i 位置，删除了 j 个数，所能满足条件的数的个数；

把这个分为两种情况

• 在 i 处删除了 0 个数，dp[i][j] = dp[i - 1][j] + (a[i] == i)；
• 在 i 处删除了 j 个数，也分为两种情况：

1. 在 i 处删除了 j - 1 个数，并且要删除 i 位置的数，dp[i][j] = dp[i - 1][j - 1];
2. 在 i 处删除了 j 个数，只需要看 i 位置的数是否满足条件，dp[i][j] = dp[i - 1][j] + (a[i] == i - j);

AC代码

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <string>
#include <algorithm>
#include <queue>
#include <utility>
#include <stack>
#include <map>
#include <vector>
#include <set>
#include <iomanip>
#include <unordered_map>
#include <deque> 
#define hz020 return
#define mes memset
#define mec memcpy
#define x first
#define y second

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll,ll> PII;

const double pi = acos(-1);
const int N = 2010;
const int null = 0x3f3f3f3f,INF = 1e9;
const ll mod = 998244353;

int T;
int n, k;
int a[N];
int dp[N][N];//在i位置已经删了j个数，所满足条件的数字个数；

int main()
{
	cin >> T;
	while (T --)
	{
		cin >> n >> k;

		for (int i = 1; i <= n; i ++ )
		{
			cin >> a[i];
			for (int j = 0; j <= i; j ++)
			{
				//当在i位置一个都不删，那么在i位置符合条件的数字个数=在i-1位置符合条件的数字个数+i位置是否符合条件
				if (!j) dp[i][j] = dp[i - 1][j] + (a[i] == i ? 1 : 0);
				//在i位置删j个，就是在i-1位置删除j-1个数，然后在i位置删一个，或者在i-1位置就删了j个数，然后看i位置是否符合条件
				else dp[i][j] = max(dp[i - 1][j - 1], dp[i - 1][j] + (a[i] == (i - j) ? 1 : 0));
			}
		}

		int res = -1;
		for (int i = 0; i <= n; i ++)
		{
			if (dp[n][i] >= k)
			{
				res = i;
				break;
			}
		}

		cout << res << endl;
	}

	hz020 0;	
}

END