This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
#include <iostream>
#include "garden.h"
#include "gardenlib.h"
#define lcm(a,b) (a/__gcd(a,b))*b
#define fast ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ll long long int
#define vi vector<int>
#define vll vector<ll>
#define pb push_back
#define F first
#define S second
#define mp make_pair
using namespace std;
const int tam=1e6;
int up[tam][21];
int up2[tam][21];
int last[tam][21];
int last2[tam][21];
vector<pair<int,int> > G[tam];
void count_routes(int N, int M, int P, int R[][2], int Q, int g[])
{
int n=N;
for(int i=0;i<M;i++){
int a=R[i][0],b=R[i][1];
if(G[a].size()<2)G[a].pb({b,i});
if(G[b].size()<2)G[b].pb({a,i});
}
for(int i=0;i<n;i++){
if(G[i].size()==0)continue;
up[i][0]=up2[i][0]=G[i][0].F;
last[i][0]=last2[i][0]=G[i][0].S;
if(G[i].size()==2){
up2[i][0]=G[i][1].F;
last2[i][0]=G[i][1].S;
}
}
//for(int i=0;i<n;i++){
// cout<<i<<" -> "<<up2[i][0]<<endl;
//}
for(int i=1;i<=20;i++){
//cout<<"i "<<i<<endl;
for(int nodo=0;nodo<n;nodo++){
int mid=up[nodo][i-1];
if(up[mid][0]!=last[nodo][i-1]){
up[nodo][i]=up[mid][i-1];
last[nodo][i]=last[mid][i-1];
}else{
up[nodo][i]=up2[mid][i-1];
last[nodo][i]=last2[mid][i-1];
}
//cout<<last[3][1]<<endl;
//------------------
if(last2[nodo][i-1]!=up[mid][0]){
up2[nodo][i]=up[mid][i-1];
last2[nodo][i]=last[mid][i-1];
}else{
up2[nodo][i]=up2[mid][i-1];
last2[nodo][i]=last2[mid][i-1];
}
}
}
//cout<<last[3][1]<<endl;
int k=g[0];
int res=0;
/*for(int i=0;i<n;i++){
cout<<"para "<<i<<endl;
for(int l=0;l<5;l++){
cout<<up[i][l]<<" ";
}
cout<<endl;
}*/
for(int i=0;i<n;i++){
int nodo=i;
int ant=-1;
//cout<<"empiezo en "<<i<<endl;
for(int l=20;l>=0;l--){
if((1<<l)&k){
if(ant!=up[nodo][0]){
ant=last[nodo][l];
nodo=up[nodo][l];
}else{
ant=last2[nodo][l];
nodo=up2[nodo][l];
}
//cout<<"subo "<<(1<<l)<<" "<<nodo<<" "<<ant<<endl;
}
}
if(nodo==P)res++;
}
answer(res);
}
/*
5 5 2
1 0
1 2
3 2
1 3
4 2
1
3 1
*/
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |