Submission #1514581


Source Code Expand

#include <iostream>
#include <vector>
#include <algorithm>
#include <utility>

using namespace std;
using llng = long long;
template<typename T> using v = vector<T>;
template<typename T> using vv = v<v<T> >;

v<int> b;
pair<int, int> loop(int a) {
    pair<int, int> ret = make_pair(0, 0);
    int r = a, t = a;
    while (true) {
        r = b[b[r]]; t = b[t];
        if (r == t) break;
    }
    r = a;
    while (true) {
        if (r == t) break;
        r = b[r]; t = b[t];
        ret.first++;
    }
    while (true) {
        r = b[r];
        ret.second++;
        if (r == t) break;
    }
    return ret;
}

int main() {
    int N, a; cin >> N >> a;
    string k; cin >> k;
    b = v<int>(N+1); for (int i=1; i<=N; i++) cin >> b[i];

    pair<int, int> p = loop(a);

    int s;
    if (k.length() < 8) {
        s = stoi(k);
        if (s > p.first) s = (s - p.first) % p.second + p.first;
    } else {
        s = 0;
        for (int i=0; i<8; i++) s = s*10 + k[i] - '0';
        s -= p.first;
        s %= p.second;
        for (int i=8; i<k.length(); i++) s = (s*10 + k[i] - '0') % p.second;
        s += p.first;
    }
    for (int i=0; i<s; i++) a = b[a];
    cout << a << endl;
    return 0;
}

Submission Info

Submission Time
Task D - へんてこ辞書
User hidollara
Language C++14 (GCC 5.4.1)
Score 0
Code Size 1265 Byte
Status WA
Exec Time 30 ms
Memory 896 KB

Judge Result

Set Name Sample Subtask1 All
Score / Max Score 0 / 0 0 / 50 0 / 50
Status
AC × 2
AC × 2
WA × 10
AC × 4
WA × 21
Set Name Test Cases
Sample subtask0_sample_01.txt, subtask0_sample_03.txt
Subtask1 subtask0_0.txt, subtask0_1.txt, subtask0_2.txt, subtask0_3.txt, subtask0_4.txt, subtask0_5.txt, subtask0_6.txt, subtask0_7.txt, subtask0_8.txt, subtask0_9.txt, subtask0_sample_01.txt, subtask0_sample_03.txt
All subtask0_0.txt, subtask0_1.txt, subtask0_2.txt, subtask0_3.txt, subtask0_4.txt, subtask0_5.txt, subtask0_6.txt, subtask0_7.txt, subtask0_8.txt, subtask0_9.txt, subtask0_sample_01.txt, subtask0_sample_03.txt, subtask1_0.txt, subtask1_1.txt, subtask1_10.txt, subtask1_11.txt, subtask1_2.txt, subtask1_3.txt, subtask1_4.txt, subtask1_5.txt, subtask1_6.txt, subtask1_7.txt, subtask1_8.txt, subtask1_9.txt, subtask1_sample_02.txt
Case Name Status Exec Time Memory
subtask0_0.txt WA 25 ms 640 KB
subtask0_1.txt WA 21 ms 512 KB
subtask0_2.txt WA 23 ms 512 KB
subtask0_3.txt WA 17 ms 512 KB
subtask0_4.txt WA 26 ms 640 KB
subtask0_5.txt WA 20 ms 512 KB
subtask0_6.txt WA 17 ms 512 KB
subtask0_7.txt WA 23 ms 512 KB
subtask0_8.txt WA 18 ms 512 KB
subtask0_9.txt WA 25 ms 640 KB
subtask0_sample_01.txt AC 1 ms 256 KB
subtask0_sample_03.txt AC 1 ms 256 KB
subtask1_0.txt WA 21 ms 640 KB
subtask1_1.txt WA 25 ms 768 KB
subtask1_10.txt WA 30 ms 896 KB
subtask1_11.txt WA 1 ms 256 KB
subtask1_2.txt AC 20 ms 640 KB
subtask1_3.txt WA 17 ms 512 KB
subtask1_4.txt WA 19 ms 640 KB
subtask1_5.txt WA 26 ms 768 KB
subtask1_6.txt WA 20 ms 640 KB
subtask1_7.txt WA 18 ms 640 KB
subtask1_8.txt WA 29 ms 768 KB
subtask1_9.txt WA 29 ms 768 KB
subtask1_sample_02.txt AC 1 ms 256 KB