Submission #2124629


Source Code Expand

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ull = unsigned long long;
#define MOD 1000000007

const int NMAX=100000;
int N,a;
string s;
int G[NMAX];
bool used[NMAX]={};

int main(){
  cin>>N>>a>>s;
  a--;
  for(int i=0;i<N;i++){
    cin>>G[i];
    G[i]--;
  }
  if(s.length()<=6){
    int k=stoi(s);
    for(int i=0;i<k;i++) a=G[a];
    cout<<a+1<<endl;
    return 0;
  }
  int cur=a,to_c=0;
  while(1){
    used[cur]=1;
    to_c++;
    cur=G[cur];
    if(used[cur]){
      break;
    }
  }
  int C=0,start=cur;
  while(1){
    C++;
    cur=G[cur];
    if(cur==start) break;
  }
  int k=0;
  for(int i=0;i<s.length();i++){
    k=(k*10+s[i]-'0')%C;
  }
  while(k<to_c){
    k+=C;
  }
  while(k>0){
    k--;
    cur=G[cur];
  }
  cout<<cur+1<<endl;
  return 0;
}

Submission Info

Submission Time
Task D - へんてこ辞書
User misosoup
Language C++14 (GCC 5.4.1)
Score 0
Code Size 845 Byte
Status WA
Exec Time 38 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 × 3
WA × 22
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 31 ms 640 KB
subtask0_1.txt WA 26 ms 640 KB
subtask0_2.txt WA 29 ms 640 KB
subtask0_3.txt WA 22 ms 512 KB
subtask0_4.txt WA 35 ms 640 KB
subtask0_5.txt WA 25 ms 640 KB
subtask0_6.txt WA 22 ms 512 KB
subtask0_7.txt WA 30 ms 640 KB
subtask0_8.txt WA 23 ms 512 KB
subtask0_9.txt WA 31 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 26 ms 768 KB
subtask1_1.txt WA 30 ms 768 KB
subtask1_10.txt WA 38 ms 896 KB
subtask1_11.txt WA 1 ms 256 KB
subtask1_2.txt WA 25 ms 640 KB
subtask1_3.txt WA 21 ms 640 KB
subtask1_4.txt WA 24 ms 640 KB
subtask1_5.txt WA 33 ms 896 KB
subtask1_6.txt WA 24 ms 768 KB
subtask1_7.txt WA 23 ms 768 KB
subtask1_8.txt WA 37 ms 896 KB
subtask1_9.txt WA 37 ms 896 KB
subtask1_sample_02.txt AC 1 ms 256 KB