zoukankan      html  css  js  c++  java
  • bzoj1984 月下“毛景树”

    Description

    毛毛虫经过及时的变形,最终逃过的一劫,离开了菜妈的菜园。 毛毛虫经过千山万水,历尽千辛万苦,最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面,发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有N个节点和N-1条树枝,但节点上是没有毛毛果的,毛毛果都是长在树枝上的。但是这棵“毛景树”有着神奇的魔力,他能改变树枝上毛毛果的个数:  Change k w:将第k条树枝上毛毛果的个数改变为w个。  Cover u v w:将节点u与节点v之间的树枝上毛毛果的个数都改变为w个。  Add u v w:将节点u与节点v之间的树枝上毛毛果的个数都增加w个。 由于毛毛虫很贪,于是他会有如下询问:  Max u v:询问节点u与节点v之间树枝上毛毛果个数最多有多少个。

    Input

    第一行一个正整数N。 接下来N-1行,每行三个正整数Ui,Vi和Wi,第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi,树枝上有Wi个毛毛果。 接下来是操作和询问,以“Stop”结束。

    Output

    对于毛毛虫的每个询问操作,输出一个答案。

    Sample Input

    4
    1 2 8
    1 3 7
    3 4 9
    Max 2 4
    Cover 2 4 5
    Add 1 4 10
    Change 1 16
    Max 2 4
    Stop

    Sample Output

    9
    16

    【Data Range】
    1<=N<=100,000,操作+询问数目不超过100,000。
    保证在任意时刻,所有树枝上毛毛果的个数都不会超过10^9个。
     
    树链剖分第二题
    写完感觉不会再爱了
    线段树写得我快跪了
    这题是边权还要转成点权处理
    #include<cstdio>
    #include<iostream>
    #include<cstring>
    #include<cstdlib>
    #include<algorithm>
    #include<cmath>
    #include<queue>
    #include<deque>
    #include<set>
    #include<map>
    #include<ctime>
    #define LL long long
    #define inf 0x7ffffff
    #define pa pair<int,int>
    #define pi 3.1415926535897932384626433832795028841971
    #define N 200010
    using namespace std;
    inline LL read()
    {
        LL x=0,f=1;char ch=getchar();
        while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
        while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
        return x*f;
    }
    int n,cnt,cnt2,tt,lca,x,y,z;
    struct edge{
        int to,next,v,rnk;
    }e[2*N];
    struct segtree{
        int l,r,mx,tag,cover;
    }tree[4*N];
    int query[N],v[N];
    int head[N];
    bool mrk[N];int depth[N],son[N],fa[N][21];
    int place[N],pplace[N],belong[N];
    char ch[100];
    inline void ins(int u,int v,int w,int rnk)
    {
        e[++cnt].to=v;
        e[cnt].v=w;
        e[cnt].next=head[u];
        e[cnt].rnk=rnk;
        head[u]=cnt;
    }
    inline void insert(int u,int v,int w,int rnk)
    {
        ins(u,v,w,rnk);
        ins(v,u,w,rnk);
    }
    inline void dfs1(int x,int dep)
    {
        if(mrk[x])return;
        mrk[x]=1;depth[x]=dep;son[x]=1;
        for(int i=1;i<=20;i++)fa[x][i]=fa[fa[x][i-1]][i-1];
        for (int i=head[x];i;i=e[i].next)
          if (!mrk[e[i].to])
          {
            fa[e[i].to][0]=x;
            query[e[i].rnk]=e[i].to;
            v[e[i].to]=e[i].v;
            dfs1(e[i].to,dep+1);
            son[x]+=son[e[i].to];
          }
    }
    inline void dfs2(int x,int chain)
    {
        place[x]=++tt;pplace[tt]=x;
        belong[x]=chain;
        int mx=-inf,res=-1;
        for (int i=head[x];i;i=e[i].next)
          if(e[i].to!=fa[x][0])
          {
            if (son[e[i].to]>mx)
            {
                mx=son[e[i].to];
                res=e[i].to;
            }
          }
        if (res==-1)return;
        dfs2(res,chain);
        for (int i=head[x];i;i=e[i].next)
          if (e[i].to!=fa[x][0]&&e[i].to!=res)
            dfs2(e[i].to,e[i].to);
    }
    inline int LCA(int a,int b)
    {
        if (depth[a]<depth[b])swap(a,b);
        int res=depth[a]-depth[b];
        for(int i=0;i<=20;i++)
          if (res & (1<<i))a=fa[a][i];
        for (int i=20;i>=0;i--)
        if (fa[a][i]!=fa[b][i])
         {
            a=fa[a][i];
            b=fa[b][i];
         }
         if (a==b)return a;
         return fa[a][0];
    }
    inline void pushdown(int k)
    {
    	int l=tree[k].l,r=tree[k].r,tag=tree[k].tag,cov=tree[k].cover;
    	tree[k].tag=0;tree[k].cover=-1;
    	if (l==r)return;
    	if (cov!=-1)
    	{
    		tree[k<<1].mx=tree[k<<1|1].mx=cov;
    		tree[k<<1].tag=tree[k<<1|1].tag=0;
    		tree[k<<1].cover=tree[k<<1|1].cover=cov;
    	}
    	if (tag)
    	{
    		tree[k<<1].mx+=tag;
    		if (tree[k<<1].cover!=-1)tree[k<<1].cover+=tag;
    		else tree[k<<1].tag+=tag;
    		tree[k<<1|1].mx+=tag;
    		if (tree[k<<1|1].cover!=-1)tree[k<<1|1].cover+=tag;
    		else tree[k<<1|1].tag+=tag;
    	}	
    }
    inline void update(int k)
    {
        tree[k].mx=max(tree[k<<1].mx,tree[k<<1|1].mx);
    }
    inline void buildtree(int now,int l,int r)
    {
        tree[now].l=l;tree[now].r=r;
        tree[now].cover=-1;
        if (l==r)
        {
            tree[now].mx=v[pplace[l]];
            return;
        }
        int mid=(l+r)>>1;
        buildtree(now<<1,l,mid);
        buildtree(now<<1|1,mid+1,r);
        update(now);
    }
    inline int ask_in_tree(int now,int x,int y)
    {
    	pushdown(now);
        int l=tree[now].l,r=tree[now].r;
        if (l==x&&r==y)return tree[now].mx;
        int mid=(l+r)>>1;
        if (y<=mid)return ask_in_tree(now<<1,x,y);
        else if(x>mid)return ask_in_tree(now<<1|1,x,y);
        return max(ask_in_tree(now<<1,x,mid),ask_in_tree(now<<1|1,mid+1,y));
    }
    inline void change_in_tree(int now,int x,int y,int dat)
    {
    	pushdown(now);
        int l=tree[now].l,r=tree[now].r;
        if (l==x&&r==y)
        {
            tree[now].mx=dat;
            tree[now].cover=dat;
            tree[now].tag=0;
            return;
        }
        int mid=(l+r)>>1;
        if (y<=mid)change_in_tree(now<<1,x,y,dat);
        else if (x>mid)change_in_tree(now<<1|1,x,y,dat);
        else
        {
            change_in_tree(now<<1,x,mid,dat);
            change_in_tree(now<<1|1,mid+1,y,dat);
        }
        update(now);
    }
    inline void add_in_tree(int now,int x,int y,int dat)
    {
    	pushdown(now);
        int l=tree[now].l,r=tree[now].r;
        if (l==x&&r==y)
        {
        	tree[now].tag+=dat;
        	tree[now].mx+=dat;
            return;
        }
        int mid=(l+r)>>1;
        if (y<=mid)add_in_tree(now<<1,x,y,dat);
        else if (x>mid)add_in_tree(now<<1|1,x,y,dat);
        else
        {
            add_in_tree(now<<1,x,mid,dat);
            add_in_tree(now<<1|1,mid+1,y,dat);
        }
        update(now);
    }
    inline int ask(int from,int to)
    {
        int l,r,mx=0;
        while (belong[from]!=belong[to])
        {
            l=place[belong[from]];
            r=place[from];
            mx=max(mx,ask_in_tree(1,l,r));
            from=fa[belong[from]][0];
        }
        l=place[to]+1;r=place[from];
        if (l>r)return mx;
        mx=max(mx,ask_in_tree(1,l,r));
        return mx;
    }
    inline void add(int from,int to,int dat)
    {
        int l,r;
        while (belong[from]!=belong[to])
        {
            l=place[belong[from]];
            r=place[from];
            add_in_tree(1,l,r,dat);
            from=fa[belong[from]][0];
        }
        l=place[to]+1;r=place[from];
        if (l>r)return;
        add_in_tree(1,l,r,dat);
    }
    inline void change(int from,int to,int dat)
    {
        int l,r;
        while (belong[from]!=belong[to])
        {
            l=place[belong[from]];
            r=place[from];
            change_in_tree(1,l,r,dat);
            from=fa[belong[from]][0];
        }
        l=place[to]+1;r=place[from];
        if (l>r)return;
        change_in_tree(1,l,r,dat);
    }
    int main()
    {
        n=read();
        for (int i=1;i<n;i++)
        {
            x=read();
            y=read();
            z=read();
            insert(x,y,z,++cnt2);
        }
        dfs1(1,1);
        dfs2(1,1);
        buildtree(1,1,n);
        while (1)
        {
            scanf("%s",ch);
            if(ch[1]=='t')break;//stop
            if (ch[1]=='a')//max
            {
                x=read();y=read();
                lca=LCA(x,y);
                printf("%d
    ",max(ask(x,lca),ask(y,lca)));
            }
            if (ch[1]=='o')//cover
            {
                x=read();y=read();z=read();
                lca=LCA(x,y);
                change(x,lca,z);change(y,lca,z);
            }
            if (ch[1]=='d')//add
            {
                x=read();y=read();z=read();
                lca=LCA(x,y);
                add(x,lca,z);add(y,lca,z);
            }
            if(ch[1]=='h')//change
            {
                x=read();y=read();
                change_in_tree(1,place[query[x]],place[query[x]],y);
            }
        }
        return 0;
    }
    

      

    ——by zhber,转载请注明来源
  • 相关阅读:
    剑指offer-树的子结构
    剑指offer-二叉搜索树的后序遍历序列
    剑指offer-调整数组顺序使奇数位于偶数前面
    剑指offer-包含min函数的栈
    剑指offer-从上往下打印二叉树
    剑指offer-链表中倒数第k个结点
    剑指offer-合并两个排列的链接
    剑指offer-替换空格
    剑指offer-旋转数组的最小数字
    剑指offer-数字在排序数组中出现的次数
  • 原文地址:https://www.cnblogs.com/zhber/p/4125675.html
Copyright © 2011-2022 走看看