Submission #463960

#	Time	Username	Problem	Language	Result	Execution time	Memory
463960		blue	Cat in a tree (BOI17_catinatree)	C++17	100 / 100	255 ms	220924 KiB

catinatree

#include <iostream>
#include <deque>
#define z size()
using namespace std; int N,D,x,q=0;deque<int> c[200'000];
deque<int> d(int u){
    deque<int> q(1,1);
    for(int v:c[u]) { auto a=d(v);
        if(q.z<a.z) swap(q,a);
        for(int i=0;i<a.z;i++)q[i]=max(q[i],max(a[i]+(D-i<q.z?q[max(D-i,i)]:0),q[i]+(D-i<a.z?a[max(D-i,i)]:0)));
        for(int i=a.z;i>=0;i--) if(i+1<q.z) q[i]=max(q[i],q[i+1]);}
    q.push_front(q.front());return q;}
int main(){
    cin>>N>>D;for(int i=1;i<N;i++){cin>>x;c[x].push_back(i);}
    for(int r:d(0))q=max(q,r);cout<<q<<'\n';}

Compilation message (stderr)

catinatree.cpp: In function 'std::deque<int> d(int)':
catinatree.cpp:9:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |         for(int i=0;i<a.z;i++)q[i]=max(q[i],max(a[i]+(D-i<q.z?q[max(D-i,i)]:0),q[i]+(D-i<a.z?a[max(D-i,i)]:0)));
      |                      ^
catinatree.cpp:9:58: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |         for(int i=0;i<a.z;i++)q[i]=max(q[i],max(a[i]+(D-i<q.z?q[max(D-i,i)]:0),q[i]+(D-i<a.z?a[max(D-i,i)]:0)));
      |                                                          ^
catinatree.cpp:9:89: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |         for(int i=0;i<a.z;i++)q[i]=max(q[i],max(a[i]+(D-i<q.z?q[max(D-i,i)]:0),q[i]+(D-i<a.z?a[max(D-i,i)]:0)));
      |                                                                                         ^
catinatree.cpp:10:39: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   10 |         for(int i=a.z;i>=0;i--) if(i+1<q.z) q[i]=max(q[i],q[i+1]);}
      |                                       ^
catinatree.cpp: In function 'int main()':
catinatree.cpp:14:5: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
   14 |     for(int r:d(0))q=max(q,r);cout<<q<<'\n';}
      |     ^~~
catinatree.cpp:14:31: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
   14 |     for(int r:d(0))q=max(q,r);cout<<q<<'\n';}
      |                               ^~~~

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