#include<bits/stdc++.h>
#include "garden.h"
#include "gardenlib.h"
using namespace std;
int a,k,n,m,q,p,v1[300005],sz,sz1,sz2,st,ch,res;
bool fx[300005];
vector<pair<int,int> >v[150005];
vector<int>rv[300005];
void cch(int x,int dis,int st){
for(vector<int> :: iterator it=rv[x].begin();it!=rv[x].end();it++){
if(*it==st) {sz=dis+1;return;}
cch(*it,dis+1,st);
}
}
void dfs(int x,int dis,int st){
if(k<dis) return;
//cout<<st<<" "<<x<<" "<<dis<<" "<<k<<" "<<sz<<endl;
if(x&1==0 && abs(k-dis)%sz==0)
{
res+=(1^fx[x]);
fx[x]=1;
}
for(vector<int> :: iterator it=rv[x].begin();it!=rv[x].end();it++){
if(*it==st) continue;
dfs(*it,dis+1,st);
}
}
void count_routes(int N, int M, int P, int R[][2], int Q, int G[])
{
int n=N,m=M,p=P;
for(int i=0;i<m;i++){///grafi1
v[R[i][0]].push_back({i,R[i][1]});
v[R[i][1]].push_back({i,R[i][0]});
}
for(int i=0;i<n;i++){
sort(v[i].begin(),v[i].end());
}
for(int i=0;i<n;i++){///grafi2
if(v[i].size()==1)
{
int to=v[i][0].second;
int val=v[i][0].first;
if(v[to][0].first==val)
v1[2*i+1]=2*to+1;
else
v1[2*i+1]=2*to;
if(v[to][0].first==val)
v1[2*i]=2*to+1;
else
v1[2*i]=2*to;
}
else
{
int to=v[i][0].second;
int val=v[i][0].first;
// cout<<i<<" "<<to<<" "<<val<<" "<<v[to][0].first<<endl;
if(v[to][0].first==val)
v1[2*i]=2*to+1;
else
v1[2*i]=2*to;
to=v[i][1].second;
val=v[i][1].first;
// cout<<i<<" "<<to<<" "<<val<<" "<<v[to][0].first<<endl;
if(v[to][0].first==val)
v1[2*i+1]=2*to+1;
else
v1[2*i+1]=2*to;
}
}
for(int i=0;i<2*n;i++){
//cout<<i<<" "<<v1[i]<<endl;
rv[v1[i]].push_back(i);
}
sz=1000000003;cch(2*p,0,2*p);sz1=sz;
sz=1000000003;cch(2*p+1,0,2*p+1);sz2=sz;
for(int i=0; i<Q; i++)
{
k=G[i];
for(int j=0;j<2*n;j++)
fx[j]=0;
res=0;
sz=sz1;dfs(2*p,0,2*p);
sz=sz2;dfs(2*p+1,0,2*p+1);
answer(res);
}
}
Compilation message
garden.cpp: In function 'void dfs(int, int, int)':
garden.cpp:22:11: warning: suggest parentheses around comparison in operand of '&' [-Wparentheses]
if(x&1==0 && abs(k-dis)%sz==0)
~^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
12 ms |
11016 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
12 ms |
11016 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
12 ms |
11016 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |