[Ipsc2015]Generating Synergy
阿新 • • 發佈:2019-01-13
傳送門
這個題有點意思啊,我可能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); } }