Submission #127393

#	Time	Username	Problem	Language	Result	Execution time	Memory
127393		impri	Birokracija (COCI18_birokracija)	C++14	100 / 100	255 ms	27000 KiB

birokracija

#include<bits/stdc++.h>
using namespace std;
int number;
vector<int>graph[200001];
int visited[200001];
int siz[200001];
long long money[200001];
int get_size(int n){
  visited[n]=1;
  int result=1;
  for(int i=0;i<graph[n].size();i++){
    if(!visited[graph[n][i]]){
        result+=get_size(graph[n][i]);
    }
  }
  return siz[n]=result;



}
long long get_m(int n){
  long long result=1;
    visited[n]=1;
    for(int i=0;i<graph[n].size();i++){
            if(!visited[graph[n][i]]){
                result+=(get_m(graph[n][i])+siz[graph[n][i]]);
            }

    }
    return money[n]=result;
}
int main() {
 cin >> number;
 for(int i=2;i<=number;i++){
    int a;
    cin >> a;
    graph[i].push_back(a);
    graph[a].push_back(i);
 }
 get_size(1);
 memset(visited,0,sizeof(visited));
 get_m(1);
 for(int i=1;i<=number;i++){
    cout << money[i] << " ";
 }

}

Compilation message (stderr)

birokracija.cpp: In function 'int get_size(int)':
birokracija.cpp:11:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<graph[n].size();i++){
               ~^~~~~~~~~~~~~~~~
birokracija.cpp: In function 'long long int get_m(int)':
birokracija.cpp:24:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<graph[n].size();i++){
                 ~^~~~~~~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10