제출 #790002

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
790002		1075508020060209tc	Cat in a tree (BOI17_catinatree)	C++14	51 / 100	1083 ms	43896 KiB

catinatree

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx,popcnt,sse4,abm")
#include <bits/stdc++.h>
using namespace std;
int n;int D;
vector<int>e[200005];
int dph[200005];
int tak[200005];
vector<int>dfn;
int rdfn[1000006];

void dfs(int nw,int pa){
dfn.push_back(nw);
rdfn[nw]=dfn.size()-1;
dph[nw]=dph[pa]+1;
for(int i=0;i<e[nw].size();i++){
    int v=e[nw][i];
    if(v==pa){continue;}
    dfs(v,nw);
    dfn.push_back(nw);
}
}
vector<int>ans;
int dis[200005];
void bfs(){
queue<int>q;
for(int i=1;i<=n;i++){
    dis[i]=1e9;
}
for(int i=0;i<ans.size();i++){
    q.push(ans[i]);
    dis[ans[i]]=0;
}
while(q.size()){
    int nw=q.front();
    q.pop();
    for(int i=0;i<e[nw].size();i++){
        int v=e[nw][i];
        if(dis[v]<=n){continue;}
        dis[v]=dis[nw]+1;
        q.push(v);
    }
}
}
int rmq[22][500005];
int lca(int a,int b){
int l=rdfn[a];int r=rdfn[b];
if(r<l){swap(l,r);}
int lg=__lg(r-l+1);
int ret=rmq[lg][l];
if(dph[ret]>dph[rmq[lg][r-(1<<lg)+1]]){
    ret=rmq[lg][r-(1<<lg)+1];
}
return ret;
}
int dst(int a,int b){
int lc=lca(a,b);
return dph[a]+dph[b]-dph[lc]*2;
}

signed main()
{
    cin.tie(0);
    ios_base::sync_with_stdio(0);
    dfn.push_back(0);
    cin>>n>>D;
    for(int i=2;i<=n;i++){
        int v;
        cin>>v;
        v++;
        e[v].push_back(i);
        e[i].push_back(v);
    }
    dfs(1,0);
    vector<pair<int,int>>seq;
    for(int i=1;i<=n;i++){
        seq.push_back({dph[i],i});
    }
    sort(seq.begin(),seq.end());
    reverse(seq.begin(),seq.end());
    for(int i=1;i<dfn.size();i++){
        rmq[0][i]=dfn[i];
    }
    for(int i=1;i<=__lg(dfn.size());i++){
        for(int j=1;j<dfn.size();j++){
            if( (j+(1<<i)-1)>=dfn.size()){break;}
            rmq[i][j]=rmq[i-1][j];
            if(dph[rmq[i][j]]>dph[rmq[i-1][j+(1<<(i-1))]]){
                rmq[i][j]=rmq[i-1][j+(1<<(i-1))];
            }
        }
    }
    int B=450;
//cout<<"hihi\n";
 //   vector<int>ans;
    ans.push_back(seq[0].second);
    tak[seq[0].second]++;
    bfs();int lst=0;
    for(int i=1;i<seq.size();i++){
        int nw=seq[i].second;
     //   cout<<nw<<" "<<dph[nw]<<endl;
        if(dis[nw]>=D){
        int ok=1;
            for(int j=i-1;j>lst;j--){
                int v=seq[j].second;
                if(tak[v]){
                    if(dst(nw,v)<D){ok=0;break;}
                }
            }
            if(ok){ans.push_back(nw);tak[nw]=1;}
        }
        if(i%450==0){
            bfs();
            lst=i;
        }
    }
    cout<<ans.size();

}

컴파일 시 표준 에러 (stderr) 메시지

catinatree.cpp: In function 'void dfs(int, int)':
catinatree.cpp:16:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   16 | for(int i=0;i<e[nw].size();i++){
      |             ~^~~~~~~~~~~~~
catinatree.cpp: In function 'void bfs()':
catinatree.cpp:30:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 | for(int i=0;i<ans.size();i++){
      |             ~^~~~~~~~~~~
catinatree.cpp:37:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |     for(int i=0;i<e[nw].size();i++){
      |                 ~^~~~~~~~~~~~~
catinatree.cpp: In function 'int main()':
catinatree.cpp:81:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   81 |     for(int i=1;i<dfn.size();i++){
      |                 ~^~~~~~~~~~~
catinatree.cpp:84:18: warning: comparison of integer expressions of different signedness: 'int' and 'long unsigned int' [-Wsign-compare]
   84 |     for(int i=1;i<=__lg(dfn.size());i++){
      |                 ~^~~~~~~~~~~~~~~~~~
catinatree.cpp:85:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   85 |         for(int j=1;j<dfn.size();j++){
      |                     ~^~~~~~~~~~~
catinatree.cpp:86:29: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   86 |             if( (j+(1<<i)-1)>=dfn.size()){break;}
      |                 ~~~~~~~~~~~~^~~~~~~~~~~~
catinatree.cpp:99:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   99 |     for(int i=1;i<seq.size();i++){
      |                 ~^~~~~~~~~~~
catinatree.cpp:93:9: warning: unused variable 'B' [-Wunused-variable]
   93 |     int B=450;
      |         ^

• Subtask 1: 11 / 11
• Subtask 2: 40 / 40
• Subtask 3: 0 / 49