题目大意
有 n n n 个人,每个人玩游戏的时间为 [ l i , r i ) [l_i, r_i) [li,ri),让你把他们分成 k k k 组,使得每一组内共同游戏时间(区间交)至少为 1,每个人必须分到一个组,每个组必须至少有一个人,让你使得所有组共同游戏时间之和最大,输出所有组最大共同游戏时间
解题思路
DP + 单调队列优化
我们将所有人按左端点排序,我们会发现有一些特别的大区间,这些区间可以完全覆盖其他区间,我们考虑这些区间,如果把它们单独拿出来,它们的贡献为区间长度,如果它们与它所包含的区间组合,不会对它包含区间的结果产生影响,所以我们考虑将这些大区间按区间长度从大到小排序,枚举
x
x
x 个大区间单独作为一组,其他区间组合起来,答案一定在其中。
我们先将大区间与其他区间分开来,将大区间求个前缀和备用。
考虑小区间
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j] 表示
i
i
i 个区间分为
j
j
j 组的最长共同游戏时间
d
p
[
i
]
[
j
]
=
m
a
x
p
∈
[
j
,
i
−
1
]
{
d
p
[
p
]
[
j
−
1
]
+
r
p
+
1
−
l
i
∩
r
p
+
1
>
l
i
}
dp[i][j] = max_{p \in [j, i-1]} \{dp[p][j-1] + r_{p+1} - l_i \ \cap \ r_{p+1} > l_i\}
dp[i][j]=maxp∈[j,i−1]{dp[p][j−1]+rp+1−li ∩ rp+1>li}
转化一下
d
p
[
i
]
[
j
]
=
−
l
i
+
m
a
x
r
p
+
1
>
l
i
{
d
p
[
p
]
[
j
−
1
]
+
r
p
+
1
}
dp[i][j] = -l_i + max_{r_{p+1} > l_i} \{dp[p][j-1] + r_{p+1} \}
dp[i][j]=−li+maxrp+1>li{dp[p][j−1]+rp+1}
我们发现,后面求的
m
a
x
max
max 是一个连续的区间,而且与
i
i
i 无关,所以我们可以用单调队列来优化这个东西,可以实现
O
(
1
)
O(1)
O(1) 的转移
然后我们就可以枚举
i
j
ij
ij 复杂度
O
(
n
∗
k
)
O(n*k)
O(n∗k)
Code
#include <bits/stdc++.h>
#define ll long long
#define qc ios::sync_with_stdio(false); cin.tie(0);cout.tie(0)
#define fi first
#define se second
#define PII pair<int, int>
#define PLL pair<ll, ll>
#define pb push_back
using namespace std;
const int MAXN = 5e3 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9 + 7;
inline int read()
{
int x=0,f=1;char ch=getchar();
while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;
}
int n, k;
int dp[MAXN][MAXN];
struct node{
int l, r;
bool operator<(node b) const{
if(l != b.l)
return l < b.l;
return r < b.r;
}
}a[MAXN], b[MAXN];
int pre[MAXN];
int v[MAXN];
int q[MAXN];
void solve(){
cin >> n >> k;
for (int i = 1; i <= n; ++i){
cin >> a[i].l >> a[i].r;
}
sort(a+1,a+1+n);
int cnt1 = 0, cnt2 = 0;
int ma = inf;
// 注意 要反着筛选
for(int i = n; i >= 1; --i){
if(a[i].r >= ma){
v[++cnt2] = a[i].r - a[i].l;
}
else{
b[++cnt1] = a[i];
ma = a[i].r;
}
}
sort(b+1, b+1+cnt1);
sort(v+1, v+1+cnt2, greater<int>());
for (int i = 1; i <= cnt2; ++i){
pre[i] = pre[i-1] + v[i];
}
int l = 1, r = 1;
memset(dp, -0x3f, sizeof dp);
dp[0][0] = 0;
for (int i = 1; i <= k; ++i){
l = 1;
r = 0;
for(int j = 1; j <= cnt1; ++j){
while(l <= r && b[q[r]].r + dp[q[r] - 1][i-1] <= dp[j-1][i-1] + b[j].r) r--;
q[++r] = j;
while(l <= r && b[q[l]].r <= b[j].l) l++;
if(l <= r)
dp[j][i] = dp[q[l] - 1][i-1] + b[q[l]].r - b[j].l;
}
}
int ans = 0;
for(int i = 0; i <= cnt2 && i < k; i++){
ans = max(ans, dp[cnt1][k-i] + pre[i]);
}
cout << ans << endl;
}
int main()
{
#ifdef ONLINE_JUDGE
#else
freopen("in.txt", "r", stdin);
freopen("out.txt", "w", stdout);
#endif
qc;
int T;
// cin >> T;
T = 1;
while(T--){
solve();
}
return 0;
}