답안 #949605

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
949605 2024-03-19T12:05:14 Z GrindMachine Sailing Race (CEOI12_race) C++17
75 / 100
3000 ms 5692 KB
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

template<typename T> using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
typedef long long int ll;
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

#define fastio ios_base::sync_with_stdio(false); cin.tie(NULL)
#define pb push_back
#define endl '\n'
#define sz(a) (int)a.size()
#define setbits(x) __builtin_popcountll(x)
#define ff first
#define ss second
#define conts continue
#define ceil2(x,y) ((x+y-1)/(y))
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define yes cout << "Yes" << endl
#define no cout << "No" << endl

#define rep(i,n) for(int i = 0; i < n; ++i)
#define rep1(i,n) for(int i = 1; i <= n; ++i)
#define rev(i,s,e) for(int i = s; i >= e; --i)
#define trav(i,a) for(auto &i : a)

template<typename T>
void amin(T &a, T b) {
    a = min(a,b);
}

template<typename T>
void amax(T &a, T b) {
    a = max(a,b);
}

#ifdef LOCAL
#include "debug.h"
#else
#define debug(x) 42
#endif

/*



*/

const int MOD = 1e9 + 7;
const int N = 500 + 5;
const int inf1 = int(1e9) + 5;
const ll inf2 = ll(1e18) + 5;

vector<int> adj[N];

void solve(int test_case)
{
    int n,k; cin >> n >> k;
    rep(u,n){
        while(true){
            int v; cin >> v;
            v--;
            if(v == -1) break;
            assert(u != v);
            adj[u].pb(v);
        }
    }

    vector<array<int,3>> order;
    rep(l,n){
        rep(r,n){
            int len = 0;
            if(l <= r){
                len = r-l+1;
            }
            else{
                len = n-l+r+1;
            }

            order.pb({len,l,r});
        }
    }

    sort(all(order));

    int dp[n][n][2];
    memset(dp,0,sizeof dp);

    auto in_range = [&](int i, int l, int r){
        if(l <= r){
            return i >= l and i <= r;
        }
        else{
            return i >= l or i <= r;
        }
    };

    for(auto [len,l,r] : order){
        if(l == r){
            rep(j,2){
                int u = 0;
                if(!j) u = l-1;
                else u = r+1;
                u = (u%n+n)%n;

                trav(v,adj[u]){
                    if(v == l){
                        dp[l][r][j] = 1;
                    }
                }
            }
        }
        else{
            rep(j,2){
                int u = 0;
                if(!j) u = l-1;
                else u = r+1;
                u = (u%n+n)%n;

                trav(v,adj[u]){
                    if(!in_range(v,l,r)) conts;

                    int val = 0;
                    if(v != l){
                        amax(val,dp[l][(v-1+n)%n][1]);
                    }
                    if(v != r){
                        amax(val,dp[(v+1)%n][r][0]);
                    }

                    val++;
                    amax(dp[l][r][j],val);
                }
            }
        }
    }

    pii ans = {0,0};

    rep(u,n){
        trav(v,adj[u]){
            int val = 0;
            if((u+1)%n != v){
                amax(val,dp[(u+1)%n][(v-1+n)%n][1]);
            }
            if((v+1)%n != u){
                amax(val,dp[(v+1)%n][(u-1+n)%n][0]);
            }

            val++;
            pii px = {val,u};
            amax(ans,px);
        }
    }

    if(!k){
        ans.ss++;
        cout << ans.ff << endl << ans.ss << endl;
        return;
    }

    vector<int> dp2(n);
    vector<bool> active(n);

    rep(u,n){
        trav(v,adj[u]){
            vector<int> order1, order2;
            for(int i = (u+1)%n; i != v; i = (i+1)%n){
                order1.pb(i);
            }
            for(int i = (v+1)%n; i != u; i = (i+1)%n){
                order2.pb(i);
            }

            if(order1.empty() or order2.empty()) conts;

            reverse(all(order1));
            order1.insert(order1.begin(),v);
            order2.insert(order2.begin(),v);

            fill(all(dp2),-inf1);
            dp2[v] = 0;
            trav(x,order1) active[x] = 1;

            trav(x,order1){
                active[x] = 0;
                trav(y,adj[x]){
                    if(active[y]){
                        amax(dp2[y],dp2[x]+1);
                    }
                }
            }

            trav(x,order2) active[x] = 1;

            trav(x,order2){
                active[x] = 0;
                trav(y,adj[x]){
                    if(active[y]){
                        amax(dp2[y],dp2[x]+1);
                    }
                }
            }

            order1.erase(find(all(order1),v));
            order2.erase(find(all(order2),v));

            trav(x,order2) active[x] = 1;

            trav(x,order1){
                trav(y,adj[x]){
                    if(active[y]){
                        int l = (v+1)%n, r = (u-1+n)%n;
                        int val = 0;
                        if(y != l){
                            amax(val,dp[l][(y-1+n)%n][1]);
                        }
                        if(y != r){
                            amax(val,dp[(y+1)%n][r][0]);
                        }

                        val += 2+dp2[x];
                        pii px = {val,u};
                        amax(ans,px);
                    }
                }
            }

            trav(x,order2) active[x] = 0;
            trav(x,order1) active[x] = 1;

            trav(x,order2){
                trav(y,adj[x]){
                    if(active[y]){
                        int l = (u+1)%n, r = (v-1+n)%n;
                        int val = 0;
                        if(y != l){
                            amax(val,dp[l][(y-1+n)%n][1]);
                        }
                        if(y != r){
                            amax(val,dp[(y+1)%n][r][0]);
                        }

                        val += 2+dp2[x];
                        pii px = {val,u};
                        amax(ans,px);
                    }
                }
            }

            trav(x,order1) active[x] = 0;
        }
    }

    ans.ss++;
    cout << ans.ff << endl << ans.ss << endl;
}

int main()
{
    fastio;

    int t = 1;
    // cin >> t;

    rep1(i, t) {
        solve(i);
    }

    return 0;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 504 KB Output is correct
2 Correct 1 ms 472 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 2 ms 344 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 9 ms 348 KB Output is correct
7 Correct 4 ms 604 KB Output is correct
8 Correct 10 ms 600 KB Output is correct
9 Correct 5 ms 604 KB Output is correct
10 Correct 12 ms 796 KB Output is correct
11 Correct 7 ms 796 KB Output is correct
12 Correct 361 ms 1368 KB Output is correct
13 Correct 546 ms 2260 KB Output is correct
14 Correct 72 ms 4560 KB Output is correct
15 Execution timed out 3064 ms 5584 KB Time limit exceeded
16 Execution timed out 3055 ms 5580 KB Time limit exceeded
17 Execution timed out 3007 ms 5560 KB Time limit exceeded
18 Correct 99 ms 5316 KB Output is correct
19 Execution timed out 3062 ms 5692 KB Time limit exceeded
20 Execution timed out 3017 ms 5560 KB Time limit exceeded