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 "garden.h"
#include "gardenlib.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
#define ll long long
#define itr ::iterator
typedef pair<int,int> pii;
const int MAX=1e6;
vector<pii> vec[MAX];
ll P_,CycleLength,Min[MAX],Distance[MAX],deg[MAX];
void dfs(int node,int type,int dis)
{
//cout<<node<<" "<<type<<" "<<dis<<" min bhi dekh lo "<<Min[node]<<"\n";
if(node==P_)
{
CycleLength=dis;
return ;
}
if(type==1)
Distance[node]=dis;
if(type==2)
{
for(auto A:vec[node])
{
if(Min[node]==A.first)
dfs(A.first,A.second,dis+1);
}
return ;
}
for(auto A:vec[node])
{
if(Min[node]==A.first and deg[node]==2)
continue;
dfs(A.first,A.second,dis+1);
}
return ;
}
void count_routes(int N, int M, int P, int R[][2], int Q, int G[])
{
P_=P;
int u,v,res=0;
for(int A=0;A<N;A++)
Distance[A]=-1,Min[A]=-1;
for(int A=0;A<M;A++)
{
u=R[A][0];
v=R[A][1];
if(deg[u]<2)
{
++deg[u];
vec[v].pb(mp(u,deg[u]));
}
if(deg[v]<2)
{
++deg[v];
vec[u].pb(mp(v,deg[v]));
}
if(Min[u]==-1)
Min[u]=v;
if(Min[v]==-1)
Min[v]=u;
}
Distance[P]=0;
for(auto A:vec[P])
dfs(A.first,A.second,1);
for(int A=0;A<N;A++)
{
if(Distance[A]==-1)
continue;
else if(CycleLength==0)
res+=Distance[A]==G[0];
//else if(G[0]-Distance[A]>=0 and (G[0]-Distance[A])%CycleLength==0)
// res++;
}
answer(res);
//cout<<res;
return ;
}
/*signed main()
{
ios_base::sync_with_stdio(false);
int G[]={3,1};
int R[][2]={{1,0},{1,2},{3,2},{1,3},{4,2}};
count_routes(5,5,2,R,2,G);
return 0;
}*/
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |