利用结构体让线段树有一个区间值的方法更新区间

来源:互联网 发布:centos 3306端口打开 编辑:程序博客网 时间:2024/05/17 08:01
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int maxn = 100000+10;

int n,sum;

struct node
{
    int l,r,n;
} a[maxn<<2];

void init(int l,int r,int i)
{
    a[i].l = l;
    a[i].r = r;
    a[i].n = 1;
    if(l!=r)
    {
        int mid = (l+r)>>1;
        init(l,mid,2*i);
        init(mid+1,r,2*i+1);
    }
}

void insert(int i,int x,int y,int m)
{
    if(a[i].n == m)//相同则不用修改了
    return ;
    if(a[i].l == x && a[i].r == y)//找到了区间,直接更新
    {
        a[i].n = m;
        return ;
    }
    if(a[i].n != -1)//如果该区间只有一种颜色
    {
        a[2*i].n = a[2*i+1].n = a[i].n;//由于后面必定对子树操作,所以更新子树的值等于父亲的值
        a[i].n = -1;//由于该区域颜色与修改不同,而且不是给定区域,所以该区域必定为杂色
    }
    //父区间为杂色时对所有子节点进行操作
    int mid = (a[i].l+a[i].r)>>1;
    if(x>mid)
    insert(2*i+1,x,y,m);
    else if(y<=mid)
    insert(2*i,x,y,m);
    else
    {
        insert(2*i,x,mid,m);
        insert(2*i+1,mid+1,y,m);
    }
}

int find(int i)//区间求和
{
    if(a[i].n != -1)//纯色直接算这个区间
    return (a[i].r - a[i].l+1)*a[i].n;
    else//不存则左右子树去找
    return find(i*2)+find(i*2+1);
}

int main()
{
    int t,i,k,x,y,m;
    int cas = 1;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&k);
        init(1,n,1);
        while(k--)
        {
            scanf("%d%d%d",&x,&y,&m);
            insert(1,x,y,m);
        }
        printf("Case %d: The total value of the hook is %d.\n",cas++,find(1));
    }

    return 0;
}
0 0
原创粉丝点击