【bzoj4636】【蒟蒻的数列】【线段树】

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

题解: 直接在线段树上打标记,最后dfs一遍整棵线段树即可.

因为区间长度比较大,所以需要动态开点.

注意一些没有建立的节点也是有贡献的,要把这些加进去.

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 40010
#define inf 1000000000
using namespace std;
int n,a,b,k,cnt,root,ls[N*40],rs[N*40],v[N*40];
long long ans;
int read(){
  int x(0);char ch=getchar();
  while (ch<'0'||ch>'9') ch=getchar();
  while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
  return x;
}
void insert(int &k,int l,int r,int ll,int rr,int x){
  if (!k) k=++cnt;
  int mid=(l+r)>>1;
  if (l==ll&&r==rr){
    v[k]=max(v[k],x);
    //cout<<l<<' '<<r<<' '<<v[k]<<endl;
    return;
  }
  if (rr<=mid) insert(ls[k],l,mid,ll,rr,x);
  else if (mid<ll) insert(rs[k],mid+1,r,ll,rr,x);
  else{
    insert(ls[k],l,mid,ll,mid,x);
    insert(rs[k],mid+1,r,mid+1,rr,x);
  }
}
void query(int k,int l,int r,int x){
  int mid=(l+r)>>1;
  if (!k) return;
  v[k]=max(v[k],x);
  if (!ls[k]&&!rs[k]){
    //cout<<l<<' '<<r<<' '<<v[k]<<endl;
    ans+=(long long)(r-l+1)*v[k];
    return;
  }
  query(ls[k],l,mid,v[k]);
  query(rs[k],mid+1,r,v[k]);
  if (!ls[k]) ans+=(long long)(mid-l+1)*v[k];
  if (!rs[k]) ans+=(long long)(r-mid)*v[k]; 
}
int main(){
  //freopen("a.in","r",stdin);
  n=read(); 
  for (int i=1;i<=n;i++){
    a=read();b=read();k=read();
    if (a==b) continue;
    insert(root,1,inf,a,b-1,k);
  }
  query(root,1,inf,0);
  cout<<ans<<endl;
}