problem
给定 mmm 条规则形如 (li,ri,ai)(l_i,r_i,a_i)(li,ri,ai),对于一个 01 串,其分数的定义是:对于第 iii 条规则,若该串在 [li,ri][l_i,r_i][li,ri] 中至少有一个 1,则该串的分数增加 aia_iai。
你需要求出长度为 nnn 的 01 串中的最大分数。
1≤n,m≤2×105,∣ai∣≤1091\le n,m\le 2\times 10^5,|a_i|\le 10^91≤n,m≤2×105,∣ai∣≤109。
solution
设 f(i,j):f(i,j):f(i,j): 到 iii 为止最后一个 111 出现位置为 jjj 的最大分数。
像这种 [l,r][l,r][l,r] 的限制,遇到线段树/树状数组,我们一般是化段为点,挂在 l/r/l−1/r+1l/r/l-1/r+1l/r/l−1/r+1 点上,具体要看题目了。
主要就是为了避免重复计算一个限制的贡献。
这里我们不妨在 rrr 的时候再加上 [l,r][l,r][l,r] 的贡献。
依次做以下两个转移:
{f(i,i)=maxj<i{f(i−1,j)}f(i,j)=f(i,j)+∑rk=i,lk≤jak\begin{cases} f(i,i)=\max_{j<i}\Big\{f(i-1,j)\Big\}\\ f(i,j)=f(i,j)+\sum_{r_k=i,l_k\le j}a_k \end{cases} {f(i,i)=maxj<i{f(i−1,j)}f(i,j)=f(i,j)+∑rk=i,lk≤jak
这种形式多半也能往线段树上靠。
具体而言:把 f(∗,j)f(*,j)f(∗,j) 放到线段树上。 jjj 做下标,表示当前最后一个 111 填在 jjj 位置的方案数。
- 对于第一种转移,直接查询线段树最大值,然后更新 iii 位置。
- 对于第二种转移,将所有 r=ir=ir=i 的 [l,r][l,r][l,r] 规则进行 [l,i][l,i][l,i] 的区间加操作。
答案即为最后的全局最大值。
code
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define maxn 200005
int n, m;
vector < pair < int, int > > g[maxn];
int Max[maxn << 2], tag[maxn << 2];
#define lson now << 1
#define rson now << 1 | 1
#define mid (l + r >> 1)
void pushdown( int now ) {tag[lson] += tag[now];tag[rson] += tag[now];Max[lson] += tag[now];Max[rson] += tag[now];tag[now] = 0;
}
void modify( int now, int l, int r, int L, int R, int v ) {if( R < l or r < L ) return;if( L <= l and r <= R ) { tag[now] += v; Max[now] += v; return; }pushdown( now );modify( lson, l, mid, L, R, v );modify( rson, mid + 1, r, L, R, v );Max[now] = max( Max[lson], Max[rson] );
}
void modify( int now, int l, int r, int p, int v ) {if( l == r ) { Max[now] = max( Max[now], v ); return; }pushdown( now );if( p <= mid ) modify( lson, l, mid, p, v );else modify( rson, mid + 1, r, p, v );Max[now] = max( Max[lson], Max[rson] );
}
signed main() {scanf( "%lld %lld", &n, &m );for( int i = 1, l, r, a;i <= m;i ++ ) {scanf( "%lld %lld %lld", &l, &r, &a );g[r].push_back( make_pair( l, a ) );}for( int i = 1;i <= n;i ++ ) {modify( 1, 0, n, i, Max[1] );for( auto it : g[i] ) modify( 1, 0, n, it.first, i, it.second );}printf( "%lld\n", Max[1] );return 0;
}