题目链接:https://codeforces.com/problemset/problem/1110/D
题意:给定n个整数和数字m(表示数字的范围),求出组成形如(i,i,i)(i, i + 1, i + 2), (i - 1, i, i + 1),(i - 2, i - 1, i)的最大个数。
题解:
注意到如果连续的三元组出现3次及以上,那么可以把他写成(i, i, i)的形式,也就是说连续三元组的个数不会超过两个,我们可以设计状态f[i][j][k]表示考虑到数字i,形如(i - 1, i, i + 1)有j个
(i, i + 1, i + 2)有k个,对于i在第三个位置(i - 2, i - 1, i)可以表示为f[i - 2]的形式。
可得方程f[i + 1][k][p] = f[i][j][k] + (cnt[i +1] - j - k - p) / 3 + p
还有一个细节就是对于(i + 1, i + 2, i + 3)来说,可能i + 1的数量不够,所以p的最大值为cnt[i + 1] - j - k.
代码:
#include <bits/stdc++.h>
#define ios ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
#define x first
#define y second
#define endl '\n'
#define int long long
using namespace std;
typedef pair<int, int> PII;
const int N = 1e6 + 10;
const int M = 2e5 + 10;
const int INF = 0x3f3f3f3f;
const double INFF = 0x7f7f7f7f7f7f7f7f;
const int mod = 1e9 + 7;
int n, m;
int a[N], f[N][3][3];
int cnt[N];
signed main()
{
ios;
cin >> n >> m;
for (int i = 1; i <= n; i++)
cin >> a[i], cnt[a[i]]++;
f[0][0][0] = 0;
for (int i = 0; i <= m + 1; i++)
{
for (int j = 0; j < 3; j++)
{
for (int k = 0; k < 3; k++)
{
int mx = cnt[i + 1] - j - k;
for (int p = 0; p < 3 && p <= mx; p++)
{
f[i + 1][k][p] = max(f[i][j][k] + (cnt[i + 1] - j - p - k) / 3 + p, f[i + 1][k][p]);
}
}
}
}
cout << f[m + 1][0][0] << endl;//小细节,f[m + 1][0][0]表示考虑到m + 1时的数量,但后面都是0,则相当于f[m]的数量
return 0;
}