码迷,mamicode.com
首页 > 其他好文 > 详细

[Ipsc2015]Generating Synergy

时间:2019-01-13 21:15:19      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:bool   for   reg   https   mes   ref   print   覆盖   tin   

传送门

这个题有点意思啊,我可能kd-tree的板子写的有点问题,查询的时候就是要暴力查左右儿子,记录的最大最小值毫无作用,不清楚为什么。
这个题也算是比较简单的啦,由于是子树操作,只要想到按dfs序建树就没有什么问题了
其他的也就是一个区间覆盖,没有什么难度
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=1e5+1,mod=1e9+7;
int cnt,T,n,c,m,ans,w,rt,id,dfn[maxn],lst[maxn],dep[maxn],pre[maxn*2],nxt[maxn*2],h[maxn];
struct oo{int d[2],mx[2],mn[2],sum,l,r,tag;}s[4*maxn];
void add(int x,int y)
{
    pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt;
    pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt;
}
void dfs(int x,int fa)
{
    dfn[x]=++id;
    for(rg int i=h[x];i;i=nxt[i])
        if(pre[i]!=fa)dep[pre[i]]=dep[x]+1,dfs(pre[i],x);
    lst[x]=id;
}
bool cmp(const oo&a,const oo&b){return a.d[w]<b.d[w];}
void update(int x)
{
    if(s[x].l)
    {
        s[x].mn[0]=min(s[x].mn[0],s[s[x].l].mn[0]);
        s[x].mn[1]=min(s[x].mn[1],s[s[x].l].mn[1]);
        s[x].mx[0]=max(s[x].mx[0],s[s[x].l].mx[0]);
        s[x].mx[1]=max(s[x].mx[1],s[s[x].l].mx[1]);
    }
    if(s[x].r)
    {
        s[x].mn[0]=min(s[x].mn[0],s[s[x].r].mn[0]);
        s[x].mn[1]=min(s[x].mn[1],s[s[x].r].mn[1]);
        s[x].mx[0]=max(s[x].mx[0],s[s[x].r].mx[0]);
        s[x].mx[1]=max(s[x].mx[1],s[s[x].r].mx[1]);
    }
}
int build(int l,int r,int d)
{
    int mid=(l+r)>>1;w=d;
    nth_element(s+l,s+mid,s+r+1,cmp);
    s[mid].mn[0]=s[mid].mx[0]=s[mid].d[0];
    s[mid].mn[1]=s[mid].mx[1]=s[mid].d[1];
    s[mid].sum=1;s[mid].tag=0;
    if(l!=mid)s[mid].l=build(l,mid-1,!d);else s[mid].l=0;
    if(r!=mid)s[mid].r=build(mid+1,r,!d);else s[mid].r=0;
    update(mid);return mid;
}
void pushdown(int x)
{
    int ls=s[x].l,rs=s[x].r;
    if(ls)s[ls].sum=s[ls].tag=s[x].tag;
    if(rs)s[rs].sum=s[rs].tag=s[x].tag;
    s[x].tag=0;
}
int get(int x,int l)
{
    if(s[x].d[0]==l)return s[x].sum;
    if(s[x].tag)pushdown(x);int ans=0;
    if(s[x].l&&l>=s[s[x].l].mn[0]&&l<=s[s[x].l].mx[0])ans=get(s[x].l,l);
    if(ans)return ans;                //这里是判断左子树是否找到的,我也不知道为什么可以找不到啊
    if(s[x].r)ans=get(s[x].r,l);    //然后找右儿子
    if(ans)return ans;                //希望大佬路过指点一下啊
}
void change(int x,int l,int r,int ll,int rr,int val)
{
    if(l<=s[x].mn[0]&&r>=s[x].mx[0]&&ll<=s[x].mn[1]&&rr>=s[x].mx[1]){s[x].sum=s[x].tag=val;return ;}
    else if(l<=s[x].d[0]&&r>=s[x].d[0]&&ll<=s[x].d[1]&&rr>=s[x].d[1])s[x].sum=val;
    if(s[x].tag)pushdown(x);
    if(s[x].l&&l<=s[s[x].l].mx[0]&&ll<=s[s[x].l].mx[1])change(s[x].l,l,r,ll,rr,val);
    if(s[x].r&&r>=s[s[x].r].mn[0]&&rr>=s[s[x].r].mn[1])change(s[x].r,l,r,ll,rr,val);
}
int main()
{
    read(T);
    while(T--)
    {
        read(n),read(c),read(m);memset(h,0,sizeof h);cnt=id=ans=0;dep[1]=1; 
        for(rg int i=2,x;i<=n;i++)read(x),add(i,x);
        dfs(1,0);for(rg int i=1;i<=n;i++)s[i].d[0]=dfn[i],s[i].d[1]=dep[i];
        rt=build(1,n,0);
        for(rg int i=1,x,y,z;i<=m;i++)
        {
            read(x),read(y),read(z);
            if(!z)ans=(ans+1ll*i*get(rt,dfn[x]))%mod;
            else change(rt,dfn[x],lst[x],dep[x]-y,dep[x]+y,z);
        }
        printf("%d\n",ans);
    }
}

[Ipsc2015]Generating Synergy

标签:bool   for   reg   https   mes   ref   print   覆盖   tin   

原文地址:https://www.cnblogs.com/lcxer/p/10263817.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!