51nod1672&HDU5700--区间交（线段树）区间枚举小技巧

题目链接https://www.51nod.com/Challenge/Problem.html#problemId=1672

小A有一个含有n个非负整数的数列与m个区间，每个区间可以表示为li,ri。

它想选择其中k个区间， 使得这些区间的交的那些位置所对应的数的和最大。（是指k个区间共同的交，即每个区间都包含这一段，具体可以参照样例）

在样例中，5个位置对应的值分别为1,2,3,4,6，那么选择[2,5]与[4,5]两个区间的区间交为[4,5]，它的值的和为10。

输入

第一行三个数n,k,m(1<=n<=100000,1<=k<=m<=100000)。
接下来一行n个数ai，表示小A的数列(0<=ai<=10^9)。
接下来m行，每行两个数li,ri，表示每个区间(1<=li<=ri<=n)。

输出

一行表示答案

输入样例

5 2 3
1 2 3 4 6
4 5
2 5
1 4

输出样例

10

---

很显然，我们可以直接枚举区间[l,r]，那么答案就是其中的某个区间。而很显然，我们可以直接枚举左端点l，然后对于在这个区间内的小区间它绝对满足L<=l并且R>=l的，也就是说对于满足条件的小区间，我们将他们扔到线段树操作，如果这个区间被大于等于k个区间覆盖，那么我们需要找到对于左端点l而言被k个区间覆盖的最大的R（这样就可以使得覆盖的元素更多和也更大），也就是找当前符合区间的最大R。那么直接用权值线段树就可以找到该位置了。但有一个问题就是我们怎么判断哪个区间满足条件，最暴力的方法就是在枚举l的同时对每个区间枚举，然后判断。。。果断T飞，实际上我们可以注意到的是，我们的l是从小到大枚举的，那么我们对每个区间的左端点排个序就好了，然后记录一下每个左端点所对应的的右端点的值，以及右端点对应的右端点的值：

for (int i=1; i<=m; i++){
      int l,r;
      scanf ("%d%d",&l,&r);
      st[l].push_back(r);ed[r].push_back(r);
}

那么我们从1到n枚举区间的时候就可以很快地判断了。我们每次枚举l的时候，在线段树中插入以l为左端点的区间（实际上只放了该区间的右端点），然后删除已l为右端点的区间及update(st[l],1),update(ed[l],-1)，这样的区间枚举就是O(n)的复杂度了，在权值线段树中找到最大的R然后用前缀和相减一下就是k个区间以l为左端点的最大元素和了。讲得不太好QAQ。。。。代码混着文字说明应该更容易理解。

以下是AC代码：

#include <bits/stdc++.h>
using namespace std;

#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define ls rt<<1
#define rs rt<<1|1
#define ll long long

const int mac=1e5+10;

int tree[mac<<2];

void update(int l,int r,int rt,int pos,int val)
{
    if (l==r){
        tree[rt]+=val;
        return;
    }
    int mid=(l+r)>>1;
    if (mid>=pos) update(lson,pos,val);
    else update(rson,pos,val);
    tree[rt]=tree[ls]+tree[rs];
}

int query(int l,int r,int rt,int val)
{
    if (l==r) return l;
    int mid=(l+r)>>1;
    if (tree[rs]>=val) return query(rson,val);
    else return query(lson,val-tree[rs]); 
}

vector<int>st[mac],ed[mac];
ll sum[mac];

int main()
{
    int n,k,m;
    while (scanf ("%d%d%d",&n,&k,&m)!=EOF){
        for (int i=1; i<=n; i++){
            st[i].clear();ed[i].clear();
            int x;
            scanf ("%d",&x);
            sum[i]+=sum[i-1]+x;
        }
        for (int i=1; i<=m; i++){
            int l,r;
            scanf ("%d%d",&l,&r);
            st[l].push_back(r);ed[r].push_back(r);
        }
        ll ans=0;
        for (int i=1; i<=n; i++){
            for (int j=0; j<st[i].size(); j++)
                update(1,n,1,st[i][j],1);
            if (tree[1]>=k){
                int pos=query(1,n,1,k);
                ans=max(ans,sum[pos]-sum[i-1]);
            }
            for (int j=0; j<ed[i].size(); j++)
                update(1,n,1,ed[i][j],-1);
        }
        memset(tree,0,sizeof tree);
        printf ("%lld\n",ans);
    }
    return 0;
}