結果

問題 No.3217 Shiki no Shiki
ユーザー dvasgdjhas
提出日時 2025-08-01 22:25:09
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,037 bytes
コンパイル時間 2,808 ms
コンパイル使用メモリ 287,300 KB
実行使用メモリ 11,952 KB
最終ジャッジ日時 2025-08-01 22:25:16
合計ジャッジ時間 4,045 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 2 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define int long long int
#define endl "\n"

void dfs(int u,int &sz,vector<bool> &vis,vector<vector<int>> &adj){
    vis[u] = 1;
    sz++;
    for(auto &v:adj[u]){
        if(vis[v]==0)dfs(v,sz,vis,adj);
    }
}

signed main(){
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int n;cin>>n;
    vector<int> p(n);
    for(int i=0;i<n;i++)cin>>p[i];
    vector<vector<int>> adj(n+1);
    vector<int> c(n+1,0);
    for(int i=0;i<n;i++){
        if(p[i]==0)continue;
        c[p[i]]++;
        adj[i+1].push_back(p[i]);
        adj[p[i]].push_back(i+1);
    }
    queue<int> q;
    for(int i=1;i<=n;i++){
        if(c[i]==0)q.push(i);
    }
    vector<bool> v(n+1,0);
    vector<int> s;
    while(!q.empty()){
        int i = q.front();
        q.pop();
        if(v[i]==0){
            int sz=0;
            dfs(i,sz,v,adj);
            s.push_back(sz);
        }
    }
    int ans=0;
    for(int i=0;i<s.size();i++){
        if(s[i]>=3)ans++;
    }
    cout<<ans;
    return 0;
}
0