oj.uz

Submission #543309

# Submission time Handle Problem Language Result Execution time Memory
543309 2022-03-30T06:23:00 Z neki Tropical Garden (IOI11_garden) C++14
100 / 100
 2669 ms 37020 KB 
#include "garden.h"
#include "gardenlib.h"
#include <bits/stdc++.h>
#define ll int
#define loop(i, a, b) for(ll i=a;i<b;++i)
#define pool(i, a, b) for(ll i=a-1;i>=b;--i)
#define fore(i, a) for(auto&& i:a)
#define fi first
#define se second
#define ps(a) push_back(a)
#define pb(a) pop_back(a)
#define eb(...) emplace_back(__VA_ARGS__)
#define sc scanf
#define vc vector
#define lb lower_bound
#define ub upper_bound
#define all(a) a.begin(), a.end()
#define llmax LLONG_MAX/2
#define llmin -LLONG_MAX/2
using namespace std;
#define par pair<ll, ll>
#define ld long double
#define mod 998244353

void count_routes(ll n, ll m, ll P,ll R[][2], ll Q, ll G[]){
    vc<vc<ll>> edg(n);
    
    loop(i, 0, m){
        edg[R[i][0]].ps(R[i][1]);
        edg[R[i][1]].ps(R[i][0]);
    }
    
    function<ll (ll, ll)> gettip=[&](ll u, ll p){return edg[u].size()>1 && edg[u][0]==p;};
    vc<vc<par>> dp(n, vc<par>(2, make_pair(-2, 0)));
    {
        vc<vc<ll>> act(n, vc<ll>(2, 0));
        ll st=P, pr=-1, cnt=0;
        ll ctip=gettip(st, pr);
        
        dp[P][ctip]=make_pair(-1, 0);
        while(1){
            ll tip=gettip(st, pr);
            
            if(act[st][tip]) break;
            act[st][tip]=1;
            
            ll nst=edg[st][tip];
            pr=st;st=nst;
            ++cnt;
            
            if(st==P){ dp[st][ctip]=make_pair(cnt, gettip(st, pr));break;}
        }
    }
    {
        vc<vc<ll>> act(n, vc<ll>(2, 0));
        ll st=P, pr=edg[P][0], cnt=0;
        ll ctip=gettip(st, pr);
        
        dp[P][ctip]=make_pair(-1, 0);
        while(1){
            ll tip=gettip(st, pr);
            
            if(act[st][tip]) break;
            act[st][tip]=1;
            
            ll nst=edg[st][tip];
            pr=st;st=nst;
            ++cnt;
            
            if(st==P){ dp[st][ctip]=make_pair(cnt, gettip(st, pr));break;}
        }
    }
    {
        vc<vc<ll>> act(n, vc<ll>(2, 0));
        function<par (ll, ll)> dfs=[&](ll u, ll p){
            ll tip=gettip(u, p);
            if(dp[u][tip].fi==-2){
                if(act[u][tip]) dp[u][tip]=make_pair(-1, 0);
                else{
                    act[u][tip]=1;
                    
                    if(edg[u][tip]==P) dp[u][tip]=make_pair(1, gettip(P, u));
                    else{ dp[u][tip]=dfs(edg[u][tip], u); if(dp[u][tip].fi!=-1) ++dp[u][tip].fi;}
                    
                    act[u][tip]=0;
                }
            }
            return dp[u][tip];
        };
        loop(u, 0, n) if(u!=P) dfs(u, -1), dfs(u, edg[u][0]);
    }
    /*loop(u, 0, n){
        loop(tip, 0, 2){cout << dp[u][tip].fi <<" "<< dp[u][tip].se<<" ";}
        cout << endl;
    }*/
    
    loop(qq, 0, Q){
        
        ll ans=0;
        loop(i, 0, n){
            ll ck=G[qq], tip;
            
            if(i==P) tip=0;
            else{
                if(dp[i][0].fi==-1) continue;
                if(ck<dp[i][0].fi) continue;
                
                ck-=dp[i][0].fi;
                tip=dp[i][0].se;
            }
            
            if(ck==0)++ans;
            else if(dp[P][tip].fi==-1);
            else if(dp[P][tip].se==tip) ans+=(ck%dp[P][tip].fi==0);
            else if(dp[P][!tip].fi==-1);
            else if(dp[P][!tip].se==tip){
                ll sk=dp[P][0].fi+dp[P][1].fi;
                if(ck%sk==0 or ck%sk==dp[P][tip].fi)++ans;
            }
            else if(dp[P][!tip].se==(!tip)) ans+=(ck>=dp[P][tip].fi && (ck-dp[P][tip].fi)%dp[P][!tip].fi==0);
        }
        answer(ans);
        //cout << ans<<endl;
    }
}

/*int main(){
    ll R[6][2]={{1, 2},{0, 1},{0, 3},{3, 4},{4, 5},{1, 5}};
    
    ll Q[1]={3};
    count_routes(6, 6, 0, R, 1, Q);
}*/

Subtask #1
49.0 / 49.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 468 KB Output is correct
2 Correct 1 ms 468 KB Output is correct
3 Correct 1 ms 468 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 596 KB Output is correct
7 Correct 1 ms 300 KB Output is correct
8 Correct 1 ms 468 KB Output is correct
9 Correct 3 ms 644 KB Output is correct

Subtask #2
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 468 KB Output is correct
2 Correct 1 ms 468 KB Output is correct
3 Correct 1 ms 468 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 596 KB Output is correct
7 Correct 1 ms 300 KB Output is correct
8 Correct 1 ms 468 KB Output is correct
9 Correct 3 ms 644 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 19 ms 4760 KB Output is correct
12 Correct 37 ms 7936 KB Output is correct
13 Correct 69 ms 29340 KB Output is correct
14 Correct 135 ms 25452 KB Output is correct
15 Correct 137 ms 25804 KB Output is correct
16 Correct 110 ms 18488 KB Output is correct
17 Correct 115 ms 15848 KB Output is correct
18 Correct 35 ms 7600 KB Output is correct
19 Correct 123 ms 25456 KB Output is correct
20 Correct 142 ms 25828 KB Output is correct
21 Correct 132 ms 18300 KB Output is correct
22 Correct 98 ms 15800 KB Output is correct
23 Correct 131 ms 28468 KB Output is correct

Subtask #3
31.0 / 31.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 468 KB Output is correct
2 Correct 1 ms 468 KB Output is correct
3 Correct 1 ms 468 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 596 KB Output is correct
7 Correct 1 ms 300 KB Output is correct
8 Correct 1 ms 468 KB Output is correct
9 Correct 3 ms 644 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 19 ms 4760 KB Output is correct
12 Correct 37 ms 7936 KB Output is correct
13 Correct 69 ms 29340 KB Output is correct
14 Correct 135 ms 25452 KB Output is correct
15 Correct 137 ms 25804 KB Output is correct
16 Correct 110 ms 18488 KB Output is correct
17 Correct 115 ms 15848 KB Output is correct
18 Correct 35 ms 7600 KB Output is correct
19 Correct 123 ms 25456 KB Output is correct
20 Correct 142 ms 25828 KB Output is correct
21 Correct 132 ms 18300 KB Output is correct
22 Correct 98 ms 15800 KB Output is correct
23 Correct 131 ms 28468 KB Output is correct
24 Correct 1 ms 316 KB Output is correct
25 Correct 96 ms 4836 KB Output is correct
26 Correct 150 ms 7956 KB Output is correct
27 Correct 1220 ms 29408 KB Output is correct
28 Correct 1102 ms 25492 KB Output is correct
29 Correct 2651 ms 25924 KB Output is correct
30 Correct 1605 ms 18524 KB Output is correct
31 Correct 1546 ms 15940 KB Output is correct
32 Correct 148 ms 7528 KB Output is correct
33 Correct 1105 ms 25488 KB Output is correct
34 Correct 2669 ms 25872 KB Output is correct
35 Correct 1681 ms 18268 KB Output is correct
36 Correct 1595 ms 15912 KB Output is correct
37 Correct 932 ms 28492 KB Output is correct
38 Correct 2212 ms 37020 KB Output is correct