[BZOJ4399] 魔法少女LJJ

题目描述

Description

在森林中见过会动的树,在沙漠中见过会动的仙人掌过后,魔法少女LJJ已经觉得自己见过世界上的所有稀奇古怪的事情了
LJJ感叹道“这里真是个迷人的绿色世界,空气清新、淡雅,到处散发着醉人的奶浆味;小猴在枝头悠来荡去,好不自在;各式各样的鲜花争相开放,各种树枝的枝头挂满沉甸甸的野果;鸟儿的歌声婉转动听,小河里飘着落下的花瓣真是人间仙境”
SHY觉得LJJ还是太naive,一天,SHY带着自己心爱的图找到LJJ,对LJJ说:“既然你已经见识过动态树,动态仙人掌了,那么今天就来见识一下动态图吧”
LJJ:“要支持什么操作?”
SHY:“
1.新建一个节点,权值为x。
2.连接两个节点。
3.将一个节点a所属于的联通快内权值小于x的所有节点权值变成x。
4.将一个节点a所属于的联通快内权值大于x的所有节点权值变成x。
5.询问一个节点a所属于的联通块内的第k小的权值是多少。
6.询问一个节点a所属联通快内所有节点权值之积与另一个节点b所属联通快内所有节点权值之积的大小。
7.询问a所在联通快内节点的数量
8.若两个节点a,b直接相连,将这条边断开。
9.若节点a存在,将这个点删去。

LJJ:“我可以离线吗?”
SHY:“可以,每次操作是不加密的,”
LJJ:“我可以暴力吗?”
SHY:“自重”
LJJ很郁闷,你能帮帮他吗

Input

第一行有一个正整数m,表示操作个数。
接下来m行,每行先给出1个正整数c。
若c=1,之后一个正整数x,表示新建一个权值为x的节点,并且节点编号为n+1(当前有n个节点)。
若c=2,之后两个正整数a,b,表示在a,b之间连接一条边。
若c=3,之后两个正整数a,x,表示a联通快内原本权值小于x的节点全部变成x。
若c=4,之后两个正整数a,x,表示a联通快内原本权值大于x的节点全部变成x。
若c=5,之后两个正整数a,k,表示询问a所属于的联通块内的第k小的权值是多少。
若c=6,之后两个正整数a,b,表示询问a所属联通快内所有节点权值之积与b所属联通快内所有节点权值之积的大小,
若a所属联通快内所有节点权值之积大于b所属联通快内所有节点权值之积,输出1,否则为0。
若c=7,之后一个正整数a,表示询问a所在联通块大小
若c=8,之后两个正整数a,b,表示断开a,b所连接的边。
若c=9,之后一个正整数a,表示断开a点的所有连边
具体输出格式见样例

Sample Input

12
1 2
1 3
1 4
1 5
1 6
2 1 2
2 2 3
2 3 4
2 4 5
9 1
3 2 5
5 3 4

Output

6

HINT

对100%的数据 0<=m<=400000,c<=7,所有出现的数均<=1000000000,所有出现的点保证存在
请认真阅读题面

题目分析

竟然c<=7 = =
那就用并查集判断连通性 每个联通块内维护权值线段树 要求支持权值线段树区间更新 线段树合并
权值之积太大怎么办? 因为 对积取log即可

#include <cstdio>
#include <cstring>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;
int m,n;
int f[400010];
int read()
{
    int sum=0;
    char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') sum=sum*10+(c-'0'),c=getchar();
    return sum;
}
int find(int x)
{ return f[x]==x?x:f[x]=find(f[x]); }
struct your
{
    int lson,rson,size;
    bool flag;
    double sum;
}a[400010*17];
int root[400010];
int tot;
void pushdown(int x)
{
    if(a[x].lson) a[a[x].lson].flag=1;
    if(a[x].rson) a[a[x].rson].flag=1;
    a[a[x].lson].sum=a[a[x].rson].sum=0;
    a[a[x].lson].size=a[a[x].rson].size=0;
    a[x].flag=0;
}
void update(int dx,int l,int r,int c,int &x)
{
    if(!x) x=++tot;
    if(l==r) 
    {
        a[x].size+=c,a[x].sum+=log(dx)*c;
        return ;
    }
    if(a[x].flag) pushdown(x);
    int mid=(l+r)>>1;
    if(dx<=mid) update(dx,l,mid,c,a[x].lson);
    else update(dx,mid+1,r,c,a[x].rson);
    a[x].size=a[a[x].lson].size+a[a[x].rson].size;
    a[x].sum=a[a[x].lson].sum+a[a[x].rson].sum;
}
void merge(int &x,int y,int l,int r)
{
    if(!x||!y) { x=x+y; return ; }
    if(l==r) 
    { 
        a[x].size+=a[y].size,a[x].sum+=a[y].sum; 
        return ;
    }
    if(a[x].flag) pushdown(x);
    if(a[y].flag) pushdown(y);
    int mid=(l+r)>>1;
    merge(a[x].lson,a[y].lson,l,mid),merge(a[x].rson,a[y].rson,mid+1,r);
    a[x].size=a[a[x].lson].size+a[a[x].rson].size;
    a[x].sum=a[a[x].lson].sum+a[a[x].rson].sum;
} 
void del(int dx,int dy,int l,int r,int x)
{
    if(!x||!a[x].size) return ;
    if(dx<=l&&r<=dy) 
    {
        a[x].size=a[x].sum=0;
        a[x].flag=1;
        return ;
    }
    if(a[x].flag) pushdown(x);
    int mid=(l+r)>>1;
    if(dy<=mid) del(dx,dy,l,mid,a[x].lson);
    else if(dx>mid) del(dx,dy,mid+1,r,a[x].rson);
    else  del(dx,mid,l,mid,a[x].lson),del(mid+1,dy,mid+1,r,a[x].rson);
    a[x].size=a[a[x].lson].size+a[a[x].rson].size;
    a[x].sum=a[a[x].lson].sum+a[a[x].rson].sum;
}
int ask(int dx,int dy,int l,int r,int x)
{
    if(dx<=l&&r<=dy) return a[x].size;
    if(a[x].flag) pushdown(x);
    int mid=(l+r)>>1;
    if(dy<=mid) return ask(dx,dy,l,mid,a[x].lson);
    else if(dx>mid) return ask(dx,dy,mid+1,r,a[x].rson);
    else return ask(dx,mid,l,mid,a[x].lson)+ask(mid+1,dy,mid+1,r,a[x].rson);
}
int find(int dx,int l,int r,int x)
{
    if(l==r) return l;
    if(a[x].flag) pushdown(x);
    int mid=(l+r)>>1;
    if(a[a[x].lson].size>=dx) return find(dx,l,mid,a[x].lson);
    else return find(dx-a[a[x].lson].size,mid+1,r,a[x].rson);
}
int main()
{
    m=read();
    for(int i=1;i<=400000;i++) f[i]=i;
    for(int tmp,x,y,c,i=1;i<=m;i++)
    {
        tmp=read();
        if(tmp==1) 
        {
            x=read();
            update(x,1,1e9,1,root[++n]);
        }
        else if(tmp==2)
        {
            x=read(),y=read(),x=find(x),y=find(y);
            if(x!=y) f[y]=x,merge(root[x],root[y],1,1e9);
        }
        else if(tmp==3)
        {
            x=read(),c=read(),x=find(x);
            y=ask(1,c,1,1e9,root[x]);
            del(1,c,1,1e9,root[x]),update(c,1,1e9,y,root[x]);
        }
        else if(tmp==4)
        {
            x=read(),c=read(),x=find(x);
            y=ask(c,1e9,1,1e9,root[x]);
            del(c,1e9,1,1e9,root[x]),update(c,1,1e9,y,root[x]);
        }
        else if(tmp==5)
            x=read(),c=read(),printf("%d\n",find(c,1,1e9,root[find(x)]));
        else if(tmp==6)
        {
            x=read(),y=read(),x=find(x),y=find(y);
            printf("%d\n",a[root[x]].sum>a[root[y]].sum);
        }
        else if(tmp==7)
            x=read(),printf("%d\n",a[root[find(x)]].size);
    }
    return 0;
}

发表评论

邮箱地址不会被公开。 必填项已用*标注