QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#859370 | #9676. Ancestors | hewanying | 0 | 0ms | 0kb | C++14 | 3.6kb | 2025-01-17 18:07:22 | 2025-01-17 18:07:23 |
answer
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
const int N=1e5+5,M=1e6+5;
int n,m,dfn[N],top[N],idx=0,dwn[N],sz[N],son[N],fa[N],dep[N],rt,ans[M],lim;
vector<int> G[N],V[N];
struct Nod{
int l,r,tim;
};
vector<Nod> C[N];
struct chg{
int y,l,r,v;
};
vector<chg> P[N];
struct Qry{
int l,x,id;
};
vector<Qry> Q[N];
void dfs1(int u){
sz[u]=1,son[u]=-1,dep[u]=dep[fa[u]]+1,lim=max(lim,dep[u]);
for(int v:G[u]){
dfs1(v);
if(!~son[u]||sz[v]>sz[son[u]]) son[u]=v;
sz[u]+=sz[v];
}
}
void dfs2(int u,int pre){
top[u]=pre,dfn[u]=++idx,dwn[pre]=u;
if(~son[u]) dfs2(son[u],pre);
for(int v:G[u]) if(v!=son[u]) dfs2(v,v);
}
void add(int x,int d){
int id=x;
P[id].pb({id,1,d-1,1});
while(x){
int p=top[x];
while(C[p].size()&&C[p].back().l<=dep[x]){
int l=C[p].back().l,r=C[p].back().r,t=C[p].back().tim;C[p].pop_back();
if(t){
if(P[id].back().y==t) P[id].back().r=d-l;
else P[id].pb({t,d-min(r,dep[x]),d-l,-1});
}
if(dep[x]<r) C[p].pb({dep[x]+1,r,t});
}
C[p].pb({dep[p],dep[x],id});
x=fa[top[x]];
}
}
void clean(int x){
while(x){
x=top[x];
C[x]={{dep[x],dep[dwn[x]],0}};
x=fa[x];
}
}
void solve(int d){
for(int i:V[d]) add(i,d);
for(int i:V[d]) clean(i);
}
namespace SGT{
#define mid ((l+r)>>1)
#define lc(p) tr[p].s[0]
#define rc(p) tr[p].s[1]
#define lson p<<1,l,mid
#define rson p<<1|1,mid+1,r
int tot=0,rt[N<<2];
struct sgt{
int s[2],v;
}tr[N*1000];
queue<int> que;
int nwnode(){
int cur=0;
if(que.size()) cur=que.front(),que.pop();
else cur=++tot;
tr[cur]={0,0,0};
return cur;
}
void upd(int &p,int x,int y,int v,int l=1,int r=lim){
if(x>y||x>r||y<l) return;
if(!p) p=nwnode();
if(x<=l&&y>=r) return tr[p].v+=v,void();
if(x<=mid) upd(lc(p),x,y,v,l,mid);
if(y>mid) upd(rc(p),x,y,v,mid+1,r);
}
int qry(int p,int x,int l=1,int r=lim){
if(!p||x<l||x>r) return 0;
if(l==r) return tr[p].v;
return (x<=mid?qry(lc(p),x,l,mid):qry(rc(p),x,mid+1,r))+tr[p].v;
}
void Upd(int y,int zl,int zr,int v,int p=1,int l=1,int r=n){
if(zl>zr||y<l||y>r) return;
upd(rt[p],zl,zr,v);
if(l==r) return;
if(y<=mid) Upd(y,zl,zr,v,lson);
else Upd(y,zl,zr,v,rson);
}
int Qry(int yl,int yr,int z,int p=1,int l=1,int r=n){
if(yl>yr||yl>r||yr<l) return 0;
if(yl<=l&&yr>=r) return qry(rt[p],z);
if(yr<=mid) return Qry(yl,yr,z,lson);
if(yl>mid) return Qry(yl,yr,z,rson);
return Qry(yl,yr,z,lson)+Qry(yl,yr,z,rson);
}
}
int main(){
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>fa[i];
if(!fa[i]) rt=i;
else G[fa[i]].pb(i);
}
dfs1(rt),dfs2(rt,rt);
for(int i=1;i<=n;i++){
V[dep[i]].pb(i);
if(top[i]==i) C[i].pb({dep[i],dep[dwn[i]],0});
}
// cerr<<"out!"<<endl;
for(int i=1;i<=n;i++) solve(i);
// cerr<<"out? "<<clock()<<"ms.\n"<<endl;
for(int i=1,l,r,x;i<=m;i++) cin>>l>>r>>x,Q[r].pb({l,x,i});
int cnt=0;
for(int i=1;i<=n;i++){
cnt+=P[i].size();
// cerr<<"i = "<<i<<endl;
for(auto j:P[i]){
// if(i>=8700) cerr<<"upd : "<<j.y<<' '<<j.l<<' '<<j.r<<" -> "<<j.v<<" -> "<<SGT::tot<<endl;
SGT::Upd(j.y,j.l,j.r,j.v);
}
// cerr<<"end : updated"<<endl;
for(auto j:Q[i]) ans[j.id]=SGT::Qry(j.l,n,j.x);
}
// cerr<<"cnt = "<<cnt<<endl;
for(int i=1;i<=m;i++) cout<<ans[i]<<'\n';
return 0;
}
详细
Subtask #1:
score: 0
Dangerous Syscalls
Test #1:
score: 0
Dangerous Syscalls
input:
7 5 3 1 0 5 3 5 1 1 3 1 5 7 2 1 5 1 4 7 1 4 7 2
output:
result:
Subtask #2:
score: 0
Skipped
Dependency #1:
0%
Subtask #3:
score: 0
Dangerous Syscalls
Test #30:
score: 0
Dangerous Syscalls
input:
50000 200000 42574 43129 47328 17982 40521 6668 12729 32377 201 11940 8599 11734 18349 41045 26854 22540 9897 33419 7463 1243 47272 27135 49050 49111 22435 42539 39924 20272 5843 9308 45963 3283 31185 13692 38952 20583 15885 24802 4773 953 49907 28689 36942 23550 19449 8970 33340 31665 5407 46023 18...
output:
result:
Subtask #4:
score: 0
Skipped
Dependency #3:
0%
Subtask #5:
score: 0
Dangerous Syscalls
Test #67:
score: 0
Dangerous Syscalls
input:
100000 1000000 6457 23693 90928 23592 90440 75018 16865 3342 83718 16731 95103 31510 38719 27886 29093 41955 6596 46409 51839 10527 91993 61074 14405 34833 53674 42363 11490 43757 46191 6058 59164 96938 57858 40178 97523 84164 21582 72243 11267 47368 97058 6637 95208 60092 53943 16441 28363 64965 52...
output:
result:
Subtask #6:
score: 0
Skipped
Dependency #1:
0%