Submission #1154362

#	Time	Username	Problem	Language	Result	Execution time	Memory
1154362		lrnnz	September (APIO24_september)	C++17	55 / 100	1093 ms	8296 KiB

september

#include <bits/stdc++.h>

using namespace std;

int solve(int N, int M, vector<int> F, vector<vector<int>> S){
    vector<vector<int>> children(N, vector<int>());
    for (int i = 1; i < F.size(); i++){
        children[F[i]].push_back(i);
    }

    int ans = 0;
    int diffs = 0;
    int minleaf = 0;

    vector<int> cnts(N, 0);
    vector<bool> visited(N, false);

    for (int i = 0; i < N-1; i++){
        for (auto x : S){
            int node = x[i];

            if (!visited[node]){
                queue<int> q;
                q.push(node);
                visited[node] = true;
                minleaf++;

                while (q.size()){
                    int newnode = q.front(); q.pop();

                    for (auto y : children[newnode]){
                        if (!visited[y]){
                            minleaf++;
                            visited[y] = true;
                            q.push(y);
                        }
                    }
                }
            }

            cnts[node]++;
            if (cnts[node] == 1){
                diffs++;
            }

            if (cnts[node] == M){
                diffs--;
            }
        }
        if (i >= minleaf-1 && diffs == 0)
        ans++;
    }

    return ans;
}

• Subtask 1: 11 / 11
• Subtask 2: 14 / 14
• Subtask 3: 5 / 5
• Subtask 4: 9 / 9
• Subtask 5: 5 / 5
• Subtask 6: 11 / 11
• Subtask 7: 0 / 9
• Subtask 8: 0 / 11
• Subtask 9: 0 / 9
• Subtask 10: 0 / 16