Submission #44275

#TimeUsernameProblemLanguageResultExecution timeMemory
44275NnandiUzastopni (COCI15_uzastopni)C++14
0 / 160
2 ms700 KiB
#include <bits/stdc++.h> using namespace std; const int maxn=10005; const int maxv=105; vector<int> graf[maxn]; bool dp[maxn][maxv][maxv]; int v[maxn]; void bejar(int start, int apa, set<int> volt) { int ez=v[start]; if(volt.count(ez)) { return; } volt.insert(ez); dp[start][ez][ez+1]=true; vector<int> dag[maxv]; dag[ez].push_back(ez+1); for(int s:graf[start]) { bejar(s,start,volt); for(int i=0;i<maxv;i++) { for(int j=i+1;j<maxv;j++) { if(dp[s][i][j]) { dag[i].push_back(j); } } } } vector<int> sor; bool volte[maxv]; for(int i=0;i<=ez;i++) { sor.resize(0); memset(volte,false,maxv); sor.push_back(i); int it=0; while(it<sor.size()) { int akt=sor[it]; it++; volte[akt]=true; for(int s:dag[akt]) { if(!volte[s]) { volte[s]=true; sor.push_back(s); } } } for(int d:sor) { if(d>ez) { dp[start][i][d]=true; } } } volt.erase(ez); return; } int main() { int n; cin>>n; for(int i=0;i<n;i++) { cin>>v[i]; } for(int i=0;i<n-1;i++) { int u, v; cin>>u>>v; u--; v--; graf[u].push_back(v); } set<int> s; bejar(0,0,s); int sol=0; for(int i=0;i<=v[0];i++) { for(int j=v[0]+1;j<maxv;j++) { if(dp[0][i][j]) { sol++; } } } cout<<sol<<endl; return 0; }

Compilation message (stderr)

uzastopni.cpp: In function 'void bejar(int, int, std::set<int>)':
uzastopni.cpp:37:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         while(it<sor.size()) {
               ~~^~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...