This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#pragma GCC optimize("O2")
#pragma GCC optimize("unroll-loops")
#include "race.h"
#include<vector>
#include<queue>
#include<array>
using namespace std;
int n,k,ans;
vector<vector<pair<int,int>>>v;
void dfs(int x,int p,long long dis,int cnt){
if(dis==k){
ans=min(ans,cnt);
return;
}
for(auto [z,w]:v[x]){
if(z!=p && dis+w<=k) dfs(z,x,dis+w,cnt+1);
}
}
// void solve(int x){
// queue<array<int,4>>q;
// q.push({x,x,0,0});
// while(!q.empty()){
// auto [x,p,dis,cnt]=q.front();
// q.pop();
// if(dis==k){
// ans=min(ans,cnt);
// continue;
// }
// for(auto [z,w]:v[x]){
// if(z!=p && dis+w<=k){
// q.push({z,x,dis+w,cnt+1});
// }
// }
// }
// }
int best_path(int N, int K, int H[][2], int L[]){
n=N;
k=K;
vector<long long>pref(n);
pref[0]=0;
v.resize(n);
bool sub1=true;
for(int i=0;i<n-1;i++){
v[H[i][0]].push_back({H[i][1],L[i]});
v[H[i][1]].push_back({H[i][0],L[i]});
pref[i+1]=pref[i]+L[i];
if(H[i][0]!=i || H[i][1]!=i+1) sub1=false;
}
ans=n;
if(sub1){
for(int i=0;i<n-1;i++){
auto pos=lower_bound(pref.begin(),pref.end(),pref[i]+K)-pref.begin();
if(pos!=n && pref[pos]-pref[i]==K) ans=min(ans,(int)pos-i);
}
return (ans==n ? -1 : ans);
}
for(int i=0;i<n;i++){
dfs(i,i,0,0);
}
return (ans==n ? -1 : ans);
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |