oj.uz

답안 #422726

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
422726 2021-06-10T11:14:55 Z ACmachine 우주 해적 (JOI14_space_pirate) C++17
0 / 100
 267 ms 524292 KB 
#include <bits/stdc++.h>
using namespace std;
 
#define FOR(i, j, k, l) for(int i = (j); i < (k); i += (l))
#define FORD(i, j, k, l) for(int i = (j); i >= (k); i -= (l))
#define REP(i, n) FOR(i, 0, n, 1)
#define REPD(i, n) FORD(i, n, 0, 1)
#define pb push_back
const int INF = (int)1e9;
typedef long long ll;

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    int n; ll k;
    cin >> n >> k;
    vector<int> g(n);
    REP(i, n){
        cin >> g[i];
        g[i]--;
    }
    vector<vector<int>> dist(n, vector<int>(n, -1)); 
    vector<vector<int>> revdist(n, vector<int>(n, -1)); // who is at dist x?
    vector<int> maxdist(n);
    function<void(int, int, int)> dfs = [&](int v, int d, int s){
        dist[s][v] = d;
        if(dist[s][g[v]] == -1){
            dfs(g[v], d + 1, s);
        }
    };
    REP(i, n){
        dfs(i, 0, i);
    } 
    REP(i, n){
        REP(j, n){
            if(dist[i][j] != -1)
                revdist[i][dist[i][j]] = j;
        }
    }
    REP(i, n){
        maxdist[i] = *max_element(dist[i].begin(), dist[i].end());
    }
    auto go = [&](int v, ll k){ // in default graph; k >= comp_size!!, otherwise doesn't work;
        if(maxdist[v] == 0) 
            return v;
        ll kk = k - maxdist[v];
        v = revdist[v][maxdist[v]];
        int u = revdist[v][1];
        ll cycle_size = 1 + dist[u][v];
        kk %= cycle_size;
        if(kk == 0) 
            return v;
        else
            return revdist[u][kk - 1]; 
    };
    vector<int> ans(n, 0);
    int def = go(0, k);
    REP(a, n){
        REP(b, n){
            if(dist[0][a] == -1){ // neviem sa dostat do zaciatku novej hrany
                ans[def]++;
            }
            else if(dist[b][a] == -1){ // nevytvorim cyklus
                ans[go(b, k - dist[0][a] - 1)]++;
            }
            else if(a == b){ // rovnake
                ans[a]++;
            } 
            else{ // vytvorim cyklus; pojdem do a
                ll kk = k - dist[0][a];
                ll cycle_size = 1 + dist[b][a];
                kk %= cycle_size;
                if(kk == 0)
                    ans[a]++;
                else
                    ans[revdist[b][kk - 1]]++; 
            } 
            //cout << a << " " << b << " " << ans[0] << "\n";
        }
    }
    REP(i, n){
        cout << ans[i] << "\n";
    }
}

Subtask #1
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 2 ms 588 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #2
0 / 37.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 2 ms 588 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #3
0 / 33.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 267 ms 524292 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 2 ms 588 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -