#include<bits/stdc++.h>
#include "garden.h"
#include "gardenlib.h"
#define INF 0x7fffffff
using namespace std;
vector<int>cg[300005],cn[300005];
int n,m,p,ed[300005][2],cur,flag,vis[300005],cnt;
int dist[2][300005],cyc_len[2];
bool state,in_cyc[2];
struct destination {
int dist[300005],cyc_len=0;
bool in_cyc=false;
}d[2];
void find_cyc(int idx) {
if(vis[idx]==1) {
if(!d[state].in_cyc && idx!=cn[p][state]) return;
d[state].in_cyc=true;
d[state].cyc_len++;
}
if(vis[idx]==2) return;
vis[idx]++;
find_cyc(cn[ed[idx][1]][(cg[ed[idx][1]][0]==ed[idx][0] && cg[ed[idx][1]].size()>1)?1:0]);
vis[idx]=0;
}
int find_dist(int idx) {
if(cn[p][state]==idx) return d[state].dist[idx]=0;
if(vis[idx]) return d[state].dist[idx];
vis[idx]++;
return d[state].dist[idx]=find_dist(cn[ed[idx][1]][(cg[ed[idx][1]][0]==ed[idx][0] && cg[ed[idx][1]].size()>1)?1:0])+1;
}
void count_routes(int N, int M, int P, int R[][2], int Q, int G[])
{
int i,j,k; n=N; m=M; p=P;
for(i=0;i<m;i++) {
ed[i][0]=R[i][0];
ed[i][1]=R[i][1];
ed[i+m][0]=R[i][1];
ed[i+m][1]=R[i][0];
cg[ed[i][0]].push_back(ed[i][1]);
cn[ed[i][0]].push_back(i);
cg[ed[i][1]].push_back(ed[i][0]);
cn[ed[i][1]].push_back(i+m);
d[0].dist[i]=-INF;
d[1].dist[i]=-INF;
d[0].dist[i+M]=-INF;
d[1].dist[i+M]=-INF;
}
state=0;
find_cyc(cn[p][0]);
state=1;
find_cyc(cn[p][1]);
state=0;
for(i=0;i<2*m;i++) {
if(vis[i])continue;
find_dist(i);
}
state=1;
memset(vis,0,sizeof(vis));
for(i=0;i<2*m;i++) {
if(vis[i])continue;
find_dist(i);
}
for(i=0;i<2;i++) {
cyc_len[i]=d[i].cyc_len;
in_cyc[i]=d[i].in_cyc;
for(j=0;j<2*m;j++) {
dist[i][j]=d[i].dist[j];
}
}
for(i=0;i<Q;i++) {
cnt=0;
for(j=0;j<n;j++) {
if(dist[0][cn[j][0]]==G[i])cnt++;
else if(dist[0][cn[j][0]]>=0 && dist[0][cn[j][0]]<G[i] && in_cyc[0] && (G[i]-dist[0][cn[j][0]])%cyc_len[0]==0)cnt++;
else if(dist[1][cn[j][0]]==G[i])cnt++;
else if(dist[1][cn[j][0]]>=0 && dist[1][cn[j][0]]<G[i] && in_cyc[1] && (G[i]-dist[1][cn[j][0]])%cyc_len[1]==0)cnt++;
}
answer(cnt);
}
}
Compilation message
garden.cpp: In function 'void count_routes(int, int, int, int (*)[2], int, int*)':
garden.cpp:37:13: warning: unused variable 'k' [-Wunused-variable]
int i,j,k; n=N; m=M; p=P;
^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
17 ms |
15728 KB |
Output is correct |
2 |
Correct |
17 ms |
15744 KB |
Output is correct |
3 |
Correct |
16 ms |
15708 KB |
Output is correct |
4 |
Correct |
16 ms |
15664 KB |
Output is correct |
5 |
Correct |
16 ms |
15660 KB |
Output is correct |
6 |
Correct |
17 ms |
15872 KB |
Output is correct |
7 |
Correct |
17 ms |
15628 KB |
Output is correct |
8 |
Correct |
19 ms |
15736 KB |
Output is correct |
9 |
Correct |
20 ms |
16376 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
17 ms |
15728 KB |
Output is correct |
2 |
Correct |
17 ms |
15744 KB |
Output is correct |
3 |
Correct |
16 ms |
15708 KB |
Output is correct |
4 |
Correct |
16 ms |
15664 KB |
Output is correct |
5 |
Correct |
16 ms |
15660 KB |
Output is correct |
6 |
Correct |
17 ms |
15872 KB |
Output is correct |
7 |
Correct |
17 ms |
15628 KB |
Output is correct |
8 |
Correct |
19 ms |
15736 KB |
Output is correct |
9 |
Correct |
20 ms |
16376 KB |
Output is correct |
10 |
Correct |
16 ms |
15580 KB |
Output is correct |
11 |
Correct |
31 ms |
18280 KB |
Output is correct |
12 |
Correct |
69 ms |
21256 KB |
Output is correct |
13 |
Correct |
85 ms |
30964 KB |
Output is correct |
14 |
Correct |
235 ms |
32280 KB |
Output is correct |
15 |
Correct |
254 ms |
32792 KB |
Output is correct |
16 |
Correct |
219 ms |
29896 KB |
Output is correct |
17 |
Correct |
221 ms |
29304 KB |
Output is correct |
18 |
Correct |
68 ms |
21272 KB |
Output is correct |
19 |
Correct |
235 ms |
32268 KB |
Output is correct |
20 |
Correct |
250 ms |
32776 KB |
Output is correct |
21 |
Correct |
242 ms |
29728 KB |
Output is correct |
22 |
Correct |
230 ms |
29084 KB |
Output is correct |
23 |
Correct |
239 ms |
33596 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
17 ms |
15728 KB |
Output is correct |
2 |
Correct |
17 ms |
15744 KB |
Output is correct |
3 |
Correct |
16 ms |
15708 KB |
Output is correct |
4 |
Correct |
16 ms |
15664 KB |
Output is correct |
5 |
Correct |
16 ms |
15660 KB |
Output is correct |
6 |
Correct |
17 ms |
15872 KB |
Output is correct |
7 |
Correct |
17 ms |
15628 KB |
Output is correct |
8 |
Correct |
19 ms |
15736 KB |
Output is correct |
9 |
Correct |
20 ms |
16376 KB |
Output is correct |
10 |
Correct |
16 ms |
15580 KB |
Output is correct |
11 |
Correct |
31 ms |
18280 KB |
Output is correct |
12 |
Correct |
69 ms |
21256 KB |
Output is correct |
13 |
Correct |
85 ms |
30964 KB |
Output is correct |
14 |
Correct |
235 ms |
32280 KB |
Output is correct |
15 |
Correct |
254 ms |
32792 KB |
Output is correct |
16 |
Correct |
219 ms |
29896 KB |
Output is correct |
17 |
Correct |
221 ms |
29304 KB |
Output is correct |
18 |
Correct |
68 ms |
21272 KB |
Output is correct |
19 |
Correct |
235 ms |
32268 KB |
Output is correct |
20 |
Correct |
250 ms |
32776 KB |
Output is correct |
21 |
Correct |
242 ms |
29728 KB |
Output is correct |
22 |
Correct |
230 ms |
29084 KB |
Output is correct |
23 |
Correct |
239 ms |
33596 KB |
Output is correct |
24 |
Correct |
18 ms |
15708 KB |
Output is correct |
25 |
Correct |
247 ms |
18416 KB |
Output is correct |
26 |
Correct |
527 ms |
21716 KB |
Output is correct |
27 |
Correct |
1694 ms |
31496 KB |
Output is correct |
28 |
Correct |
4936 ms |
32468 KB |
Output is correct |
29 |
Execution timed out |
5017 ms |
32816 KB |
Time limit exceeded |
30 |
Halted |
0 ms |
0 KB |
- |