题目链接:https://ac.nowcoder.com/acm/contest/11253/G
题目大意:
给定 n 个区间,要求将它们分成 k 组,每组之间有交,最大化每组交长度之和
分析:
区间之间的关系可以分为三类:
- 完全不相交
- 包含
- 部分相交
明显,完全不相交的区间是绝对不会分到一组里边的。
考虑包含和部分相交部分,考虑存在包含关系的两个区间,较长的区间有两种处理方式:
- 单独放到一组里边,此时,对答案的贡献就是区间长
- 放到较短区间一组内,不影响答案
由上,可以先把存在包含关系的长区间都要单独拿出来,剩下的就是部分相交或完全不相交的关系。
考虑处理剩下这部分的区间,n=5000,n^2 即可。
定义 DP[i][j] 表示把前 j 个区间分为 i 组的最大答案,那么区间转移状态为:
DP[i][j] = DP[i-1][k] + r[k] - l[j]
这里 DP[i-1][k] + r[k] 是 max(DP[i-1][1] + r[1] , ... , DP[i-1][j-1] + r[j-1] ),并且 r[k] > l[j]
显然,这是一个 n3 的转移方法(一个大佬说可以过),不过,明显第三维可以通过单调队列维护出来
维护出分组结果后,再同刚刚选出来单独的区间取组合取max即可
细节看代码吧
AC代码
#include <bits/stdc++.h>
#include <stdio.h>
#include <vector>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <stack>
#define ll long long
#define chushi(a, b) memset(a, b, sizeof(a))
#define endl "\n"
const int maxn = 5e3+5;
using namespace std;
typedef struct Node{
int l;
int r;
} node;
bool cmp(node A, node B){
return A.l == B.l ? (A.r < B.r) : (A.l < B.l);
}
node a[maxn];
ll b[maxn], cnt = 0;
int q[maxn], l, r;
ll f[maxn][maxn];
int main(){
int n, k;
cin >> n >> k;
for(int i = 1; i <= n; i++) cin >> a[i].l >> a[i].r;
sort(a+1, a+1+n, cmp);
int mn = 2e9;
for(int i = n; i >= 1; i--){ // 取出来包含关系的区间
if(a[i].r >= mn){
b[++cnt] = a[i].r - a[i].l;
a[i].l = 2e9;
}
else mn = a[i].r;
}
sort(a+1, a+1+n, cmp);
chushi(f, 0xc0); f[0][0] = 0; // 初始没有达到的状态为负无穷
n -= cnt;
for(int i = 1; i <= min(n, k); i++){
l = r = 0; q[++r] = 0; // 维护一个递减队列
for(int j = 1; i <= n; j++){
while(l<r && a[q[l+1]+1].r <= a[j].l) ++l; // 弹出和第j个区间没有相交的部分
if(l < r) f[i][j] = f[i-1][q[l+1]] + a[q[l+1]+1].r - a[j].l; // 如果有相交的区间,就维护新分组
if(j == n) break;
while(l<r && f[i-1][j] + a[j+1].r >= f[i-1][q[r]] + a[q[r]+1].r) --r; // 维护递减队列,弹掉队列中较小的元素
q[++r] = j;
}
}
sort(b+1, b+1+cnt);
reverse(b+1, b+1+cnt); // 单独的区间排序,反过来,让大的在前
ll ans = 0;
for(int i = 2; i <= cnt; i++) b[i] += b[i-1]; // 求一个前缀和
for(int i = 1; i <= min(n, k); i++) ans = max(ans, f[i][n] + b[max(k-i, 0)]); // 取max
cout << ans << endl;
return 0;
}