BZOJ 4636: 蒟蒻的数列 分块

4636: 蒟蒻的数列

题目连接：

http://www.lydsy.com/JudgeOnline/problem.php?id=4636

Description

蒟蒻DCrusher不仅喜欢玩扑克，还喜欢研究数列
题目描述
DCrusher有一个数列，初始值均为0，他进行N次操作，每次将数列[a,b)这个区间中所有比k小的数改为k，他想知
道N次操作后数列中所有元素的和。他还要玩其他游戏，所以这个问题留给你解决。

Input

第一行一个整数N，然后有N行，每行三个正整数a、b、k。
N<=40000 , a、b、k<=10^9

Output

一个数，数列中所有元素的和

Sample Input

4

2 5 1

9 10 4

6 8 2

4 6 3

Sample Output

16

Hint

题意

题解：

这种瞎JB更新的，感觉还是分块简单一点……

这道题要离散化，我偷了个懒，每个询问扔了三个点进去

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+7;
int n,m,num,belong[maxn],block,l[maxn],r[maxn],low[maxn];
vector<int> V;
map<int,int> H;
map<int,int> H2;
int A[maxn],B[maxn],K[maxn],a[maxn];
void build()
{
    block=sqrt(n/2);
    num=n/block;if(n%block)num++;
    for(int i=1;i<=num;i++)
        l[i]=(i-1)*block+1,r[i]=i*block;
    r[num]=n;
    for(int i=1;i<=n;i++)
        belong[i]=(i-1)/block+1;
}
void update(int L,int R,int k)
{
    if(R<L)return;
    if(belong[L]==belong[R])
    {
        for(int i=l[belong[L]];i<=r[belong[L]];i++)
            a[i]=max(a[i],low[belong[L]]);
        low[belong[L]]=0;
        for(int i=L;i<=R;i++)
            a[i]=max(a[i],k);
        return;
    }
    for(int i=l[belong[L]];i<=r[belong[L]];i++)
        a[i]=max(a[i],low[belong[L]]);
    low[belong[L]]=0;
    for(int i=l[belong[R]];i<=r[belong[R]];i++)
        a[i]=max(a[i],low[belong[R]]);
    low[belong[R]]=0;
    for(int i=L;i<=r[belong[L]];i++)
        a[i]=max(a[i],k);
    for(int i=l[belong[R]];i<=R;i++)
        a[i]=max(a[i],k);
 
    for(int i=belong[L]+1;i<belong[R];i++)
        low[i]=max(low[i],k);
}
 
int main()
{
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&A[i],&B[i],&K[i]);
        V.push_back(A[i]);
        V.push_back(B[i]);
        V.push_back(B[i]-1);
    }
    n=3*m;
    build();
    sort(V.begin(),V.end());
    V.erase(unique(V.begin(),V.end()),V.end());
    for(int i=0;i<V.size();i++)
        H[V[i]]=i+1,H2[i+1]=V[i];
    long long ans = 0;
    for(int i=1;i<=m;i++)
        update(H[A[i]],H[B[i]-1],K[i]);
    for(int i=0;i<V.size();i++)
        a[i+1]=max(a[i+1],low[belong[i+1]]);
    for(int i=0;i<V.size()-1;i++)
        ans+=(H2[i+2]-H2[i+1])*a[i+1];
    printf("%lld\n",ans);
}