• 洛谷 P1438 无聊的数列 线段树板子


    博客数据丢失了一点,还好有备份

    题目传送门

    题目大意:

    维护一个数列,支持两种操作:

    1.把区间([l,r])依次加上一个给定首项(K)与公差(D)等差数列。(区间修改)

    2.询问数列第(p)个数的值。(单点查询)


    线段树复苏练习好题。

    两个明显的操作显然用线段树来解决。

    主要的问题在于懒标记。

    考虑对每个节点存储两个标记((f1,f2)),表示这个区间未传给下层节点的等差数列的首项公差

    为什么可以这样表示?在线段树中,只有递归到的区间完全在需要操作的区间中,我们才会修改它的值并打上懒标记,所以对于每一次对于区间的修改,都是完整的等差数列,并且对于不同的等差数列,我们只需要把它们的首项和公差记录下来,就可以了。

    比如:({(1,3,5,7,9)}+{(1,4,7,10,13)}={(2,7,12,17,22)}),首项和公差的变化:((1,2)+(1,3)=(2,5))

    那么对于(down ext{(下沉}))操作,我们就分别对于左右儿子进行讨论。

    我们假设是这样一个标记:

    (中间为左右儿子的分界点,左右儿子分别用(ls,rs)表示)

    那么,左儿子所需要加上的等差数列的首项,就是(f1)

    而右二子所需要加上等差数列的首项,已经经历了左儿子区间长度次数的增长,即(f1+(ls.r-ls.l)*f2)

    两者的公差都为(f2)

    那么直接写肯定是不方便的(指我改了两个小时没改出来的sb错误

    所以考虑写一个函数来代劳,把((f1,f2))标记打到节点(k)

    void f(int k,int f1,int f2)
    {
    	t[k].f1+=f1,t[k].f2+=f2;
    	t[k].w+=(f1+f1+(t[k].r-t[k].l)*f2)*(t[k].r-t[k].l+1)/2;
    }
    

    下沉操作经过上面简单讨论,就变得很清晰了。

    void down(int k)
    {
    	f(ls,t[k].f1,t[k].f2);
    	f(rs,t[k].f1+(t[rs].l-t[k].l)*t[k].f2,t[k].f2);
    	t[k].f1=t[k].f2=0;
    }
    

    单点修改板子,区间修改板子。

    于是这道题就水过去啦!

    我们线段树真的太厉害啦!

    完整代码:

    #include<bits/stdc++.h>
    #define ls (k<<1)
    #define rs (k<<1|1)
    using namespace std;
    const int N=1e5+10;
    struct node
    {
    	int l,r,w;
    	int f1,f2;
    }t[N<<2];
    int x,y,kkk,d,p;
    void build(int k,int l,int r)
    {
    	t[k].l=l,t[k].r=r;
    	if(l==r)
    	{
    		scanf("%d",&t[k].w);
    		return;
    	}
    	int m=l+r>>1;
    	build(ls,l,m);
    	build(rs,m+1,r);
    	t[k].w=t[ls].w+t[rs].w;
    }
    void f(int k,int f1,int f2)
    {
    	t[k].f1+=f1,t[k].f2+=f2;
    	t[k].w+=(f1+f1+(t[k].r-t[k].l)*f2)*(t[k].r-t[k].l+1)/2;
    }
    void down(int k)
    {
    	f(ls,t[k].f1,t[k].f2);
    	f(rs,t[k].f1+(t[rs].l-t[k].l)*t[k].f2,t[k].f2);
    	t[k].f1=t[k].f2=0;
    }
    void add(int k)
    {
    	int l=t[k].l,r=t[k].r;
    	if(x<=l&&r<=y)
    	{
    		t[k].w+=((2*(l-x)*d+kkk+kkk+(r-l)*d)*(r-l+1)/2);
    		t[k].f1+=(l-x)*d+kkk;
    		t[k].f2+=d;
    		return;
    	}
    	down(k);
    	int m=l+r>>1;
    	if(x<=m)add(ls);
    	if(y>m)add(rs);
    	t[k].w=t[ls].w+t[rs].w;
    }
    int ask(int k)
    {
    	if(t[k].l==t[k].r)
    	{
    		return t[k].w;
    	}
    	down(k);
    	int m=(t[k].l+t[k].r)>>1;
    	if(p<=m)return ask(ls);
    	else return ask(rs);
    	t[k].w=t[ls].w+t[rs].w;
    }
    int main()
    {
    	int n,m;
    	cin>>n>>m;
    	build(1,1,n);
    	while(m--)
    	{
    		int op;
    		scanf("%d",&op);
    		if(op==1)
    		{
    			scanf("%d%d%d%d",&x,&y,&kkk,&d);
    			add(1);
    		}
    		else
    		{
    			scanf("%d",&p);
    			printf("%d
    ",ask(1));
    		}
    	}
    	return 0;
    }
    
  • 相关阅读:
    shell 重启 tomcat 脚本
    shell 复制/备份文件 脚本
    在 CentOS 上安装 node.js
    架构漫谈(一):什么是架构? -王概凯
    冷静审视人工智能技术的本质 | 一图看懂新一代人工智能知识体系大全
    时代在变
    什么是设计思维Design Thinking——风靡全球的创造力培养方法
    金融即服务(FaaS),将开启场景化金融新格局
    devops工具
    京东金融-供应链金融业务介绍
  • 原文地址:https://www.cnblogs.com/moyujiang/p/13383481.html
Copyright © 2020-2023  润新知