时间复杂度 O(n2k2) 的四维 dp 只需要经过简单的优化就能通过:
#include <bits/stdc++.h>
using namespace std;
int n, k;
array<pair<int, int>, 505> a;
array<array<int, 105>, 505> dp;
inline int dis(pair<int, int>& x, pair<int, int>& y) {//注意这里将函数内联
return abs(x.first - y.first) + abs(x.second - y.second);
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n >> k;
for (int i = 1; i <= n; ++i) {
cin >> a[i].first >> a[i].second;
}
sort(a.begin() + 1, a.begin() + n + 1);
for (int i = 1; i <= n; ++i) {
for (int j = 0; j <= k; ++j) {
dp[i][j] = j + 1;
}
}
for (int i = 1; i <= n; ++i) {
for (int j = 0; j <= k; ++j) {
for (int l = 1; l < i; ++l) {
for (int m = 0; m <= j; ++m) {
if (a[i].first >= a[l].first && a[i].second >= a[l].second && dis(a[i], a[l]) <= j - m + 1) {
dp[i][j] = max(dp[i][j], dp[l][m] + 1 + j - m);
}
}
}
}
}
int ans = 0;
for (int i = 1; i <= n; ++i) {
for (int j = 0; j <= k; ++j) {
ans = max(ans, dp[i][j]);
}
}
cout << ans;
return 0;
}
在 dis 函数前的 inline 显著减小了时间消耗,具体优化约有数百 ms。