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
题解:
应该是线段树吧。。
原来的写法行不通了,换了动态开节点。。
只要在树上打个标记,最后遍历一遍出解。。。
#include<stdio.h> #include<iostream> using namespace std; const int N=400005; #define ll long long int n,i,x,y,z,rt,tot,t[N<<2],ls[N<<2],rs[N<<2]; ll ans; inline void read(int &v){ char ch,fu=0; for(ch=‘*‘; (ch<‘0‘||ch>‘9‘)&&ch!=‘-‘; ch=getchar()); if(ch==‘-‘) fu=1, ch=getchar(); for(v=0; ch>=‘0‘&&ch<=‘9‘; ch=getchar()) v=v*10+ch-‘0‘; if(fu) v=-v; } inline void update(int l,int r,int x,int y,int z,int&p) { if(!p) p=++tot; if(x<=l&&r<=y) { t[p]=max(t[p],z); return; } int mid=(l+r)>>1; if(x<=mid) update(l,mid,x,y,z,ls[p]); if(y>mid) update(mid+1,r,x,y,z,rs[p]); } inline void solve(int l,int r,int p) { if(!ls[p]&&!rs[p]) { ans+=(ll)(r-l+1)*t[p]; return; } if(ls[p]) t[ls[p]]=max(t[ls[p]],t[p]); if(rs[p]) t[rs[p]]=max(t[rs[p]],t[p]); int mid=(l+r)>>1; if(ls[p]) solve(l,mid,ls[p]); if(rs[p]) solve(mid+1,r,rs[p]); if(!ls[p]) ans+=(ll)(mid-l+1)*t[p]; if(!rs[p]) ans+=(ll)(r-mid)*t[p]; } int main() { read(n); for(i=1;i<=n;i++) { read(x),read(y),read(z); y--; if(x<=y) update(1,1000000000,x,y,z,rt); } solve(1,1000000000,rt); cout<<ans; return 0; }
时间: 2024-12-11 17:28:16