题意:你有n个俄罗斯套娃,已知每个套娃的容积和体积,问有多少个子集满足以下条件:
1:这个子集是一个极大子集,即不能再添加其它的套娃到这个子集里。
2:子集的套娃之间的间隙和最小。
思路1:线段树优化DP:
首先把套娃按照容积为第一优先级,体积为第二优先级,从小到大排序。设ans[i].second为第i个套娃在最外层,间隙最小的极大子集的数目,ans[i].first为最小的间隙,我们来执行转移:假设第i个套娃的体积是r,容积比r大的第一个套娃的容积是l1, 体积是r1, 那么容易发现,所有容积在[l1, r1 - 1]范围内的套娃都可能由当前状态转移。我们可以通过线段树的特殊标记来执行转移,我在线段树维护了3个标记:容积,最小间隙,数目。在更新懒标记时,判断下放的最小间隙 - 容积是否比当前的小即可。这样通过线段树的单点查询就可以得到对应的DP状态。
代码:
#include <bits/stdc++.h>
#define LL long long
#define pLL pair<LL, LL>
#define pii pair<int, int>
#define ls (o << 1)
#define rs (o << 1 | 1)
using namespace std;
const int maxn = 200010;
const LL mod = 1000000007;
pii a[maxn];
int val[maxn];
struct Seg {
pLL lz;
int pos;
};
Seg tr[maxn * 4];
pLL ans[maxn];
bool v[maxn];
void maintain(int o, pLL tmp, int p) {
if(tmp.first - p < tr[o].lz.first - tr[o].pos) {
tr[o].lz = tmp, tr[o].pos = p;
} else if(tmp.first - p == tr[o].lz.first - tr[o].pos) {
tr[o].lz.second = (tr[o].lz.second + tmp.second) % mod;
}
}
void pushdown(int o) {
if(tr[o].lz.second != 0) {
maintain(ls, tr[o].lz, tr[o].pos);
maintain(rs, tr[o].lz, tr[o].pos);
tr[o].lz = make_pair(1e18, 0ll);
tr[o].pos = 0;
}
}
void build(int o, int l, int r) {
if(l == r) {
if(val[l]) {
tr[o].lz = make_pair(0, val[l]);
tr[o].pos = 0;
} else {
tr[o].lz = make_pair(1e18, 0);
tr[o].pos = 0;
}
return;
}
tr[o].lz = make_pair(1e18, 0);
tr[o].pos = 0;
int mid = (l + r) >> 1;
build(ls, l, mid);
build(rs, mid + 1, r);
}
pLL query(int o, int l, int r, int p) {
if(l == r) {
ans[l] = make_pair(tr[o].lz.first + a[l].first - tr[o].pos, tr[o].lz.second);
return ans[l];
}
pushdown(o);
int mid = (l + r) >> 1;
if(p <= mid) return query(ls, l, mid, p);
else return query(rs, mid + 1, r, p);
}
void update(int o, int l, int r, int ql, int qr, int p, pLL tmp) {
if(l >= ql && r <= qr) {
maintain(o, tmp, p);
return;
}
pushdown(o);
int mid = (l + r) >> 1;
if(ql <= mid) update(ls, l, mid, ql, qr, p, tmp);
if(qr > mid) update(rs, mid + 1, r, ql, qr, p, tmp);
}
int main() {
int n;
LL mi = 1e18;
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%d%d", &a[i].second, &a[i].first);
}
sort(a + 1, a + 1 + n);
for (int i = 1; i <= n; i++) {
if(a[i].first >= mi) break;
val[i] = 1;
mi = min(mi, (LL)a[i].second);
}
build(1, 1, n);
mi = 1e18;
for (int i = 1; i <= n; i++) {
pLL tmp = query(1, 1, n, i);
ans[i] = tmp;
int pos1 = lower_bound(a + 1, a + 1 + n, make_pair(a[i].second, a[i].second)) - a;
int pos2 = lower_bound(a + 1, a + 1 + n, make_pair(a[pos1].second, a[pos1].second)) - a - 1;
if(pos1 <= n)
update(1, 1, n, pos1, pos2, a[i].second, tmp);
if(pos1 <= n) {
v[i] = 1;
}
}
for (int i = 1; i <= n; i++) {
if(v[i]) continue;
mi = min(mi, ans[i].first);
}
LL res = 0;
for (int i = 1; i <= n; i++) {
if(v[i]) continue;
if(ans[i].first == mi) res = (res + ans[i].second) % mod;
}
printf("%lld\n", res);
}
思路2:(来自PinkRabbit)刚才的做法虽然能过,但是对题目的性质发掘的不够完全。我们把思维转变一下:我们不妨看一下有哪些状态可能转移到当前状态,容易发现在排序之后能转移到当前状态的状态是一个前缀,那么我们在转移的时候可以采用2个指针这种做法。那么新的问题来了,怎么知道这些前缀状态中哪些是可以真正转移到当前状态呢?设f[i]为到第i个套娃的最小间隙,ans[i]为最小间隙的数目,a[i]为第i个套娃的体积,b[i]为容积,那么在那些前缀最状态中(假设是状态t),f[t] - a[t]最小的状态才可以像当前状态转移。所以在两个指针的过程中只需维护f[t] - a[t]的最小值即可,然后执行转移。
代码:
#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int maxn = 200010;
const LL mod = 1000000007;
int a[maxn], b[maxn], p1[maxn], p2[maxn];
LL ans[maxn];
int f[maxn];
bool cmp1(int x, int y) {
return a[x] < a[y];
}
bool cmp2(int x, int y) {
return b[x] < b[y];
}
int main() {
int n, ed = -1;
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%d%d", &a[i], &b[i]);
p1[i] = p2[i] = i;
ed = max(ed, b[i]);
}
sort(p1 + 1, p1 + 1 + n, cmp1);
sort(p2 + 1, p2 + 1 + n, cmp2);
int pos = 1;
int mi = 0, sum = 1;
for (int i = 1; i <= n; i++) {
int now = p2[i];
while(pos <= n && a[p1[pos]] <= b[now]) {
int t = p1[pos];
if(f[t] - a[t] < mi) {
mi = f[t] - a[t];
sum = ans[t];
} else if(f[t] - a[t] == mi) {
sum = (sum + ans[t]) % mod;
}
pos++;
}
f[now] = b[now] + mi;
ans[now] = sum;
}
LL res = 0;
mi = 1e9;
for (int i = 1; i <= n; i++) {
if(a[i] <= ed) continue;
if(f[i] < mi) {
mi = f[i];
res = ans[i];
} else if(f[i] == mi) {
res = (res + ans[i]) % mod;
}
}
printf("%lld\n", res);
}