Submission #997420

#	Time	Username	Problem	Language	Result	Execution time	Memory
997420		Trisanu_Das	September (APIO24_september)	C++17	45 / 100	134 ms	37736 KiB

september

#include "september.h"
#include <bits/stdc++.h>
#define INF 1e18
using namespace std;
typedef long long ll;

vector<int> adj[800005];
int dp[800005];

void dfs(int s, int e) {
    for (auto u : adj[s]) {
        if (u == e) continue;
        dfs(u, s);
        dp[s] = max(dp[s], dp[u]);
    }
}

int solve(int N, int M, vector<int> F, vector<vector<int>> S) {
    for (int i = 0; i < N; i++) {
        adj[i].clear();
        dp[i] = 0;
    }
    for (int i = 1; i < N; i++) {
        adj[i].push_back(F[i]);
        adj[F[i]].push_back(i);
    }
    for (int i = 0; i < S[0].size(); i++) dp[S[0][i]] = i;
    dfs(0, -1);
    int ans = 0, i = 0;
    while (i < S[0].size()) {
        int j = dp[S[0][i]];
        int maxn = dp[S[0][i]];
        int k = i;
        while (k <= maxn) {
            maxn = max(maxn, dp[S[0][k]]);
            ++k;
        }
        ++ans;
        i = maxn + 1;
    }
    return ans;
}

Compilation message (stderr)

september.cpp: In function 'int solve(int, int, std::vector<int>, std::vector<std::vector<int> >)':
september.cpp:27:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |     for (int i = 0; i < S[0].size(); i++) dp[S[0][i]] = i;
      |                     ~~^~~~~~~~~~~~~
september.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 |     while (i < S[0].size()) {
      |            ~~^~~~~~~~~~~~~
september.cpp:31:13: warning: unused variable 'j' [-Wunused-variable]
   31 |         int j = dp[S[0][i]];
      |             ^

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