oj.uz

답안 #583538

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
583538 2022-06-25T14:20:09 Z AGE Političari (COCI20_politicari) C++14
65 / 70
 58 ms 3036 KB 
#include<bits/stdc++.h>
#define F first
#define S second
#define pb push_back
#define int long long

using namespace std;
const int N=1e6,M=2e3,mod=1e9+7;
int a[510][510];
map<pair<int,int>,int>vis;
main()
{
    int n,m;
    cin>>n>>m;

    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            cin>>a[i][j];

    vector<pair<int,int>>v;
    v.pb({0,1});
    v.pb({1,2});

    vis[{0,1}]=1;
    vis[{1,2}]=1;

    pair<int,int>ans;
    while(1){

        int x=v.size()-1;

        if(vis[{v[x].S,a[v[x].S-1][v[x].F-1]}]==1){
            ans.F=v[x].S;
            ans.S=a[v[x].S-1][v[x].F-1];
            break;
        }


        vis[{v[x].S,a[v[x].S-1][v[x].F-1]}]=1;
        v.pb({v[x].S,a[v[x].S-1][v[x].F-1]});

    }


    reverse(v.begin(),v.end());
    while((v.size())&&((*v.rbegin()).F!=ans.F||(*v.rbegin()).S!=ans.S)&&m>0)
        v.pop_back(),m--;


    reverse(v.begin(),v.end());

    int xx=m%v.size();

    if(xx==0)
        xx=v.size();

    xx--;

    cout<<v[xx].S<<endl;
    return 0;
}

Compilation message

politicari.cpp:11:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   11 | main()
      | ^~~~

Subtask #1
65.0 / 70.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 10 ms 1360 KB Output is correct
3 Correct 34 ms 2352 KB Output is correct
4 Correct 42 ms 2668 KB Output is correct
5 Correct 56 ms 3036 KB Output is correct
6 Correct 58 ms 2932 KB Output is correct
7 Incorrect 1 ms 340 KB Output isn't correct
8 Correct 4 ms 852 KB Output is correct
9 Correct 12 ms 1372 KB Output is correct
10 Correct 38 ms 2636 KB Output is correct
11 Correct 56 ms 2920 KB Output is correct
12 Correct 50 ms 2848 KB Output is correct
13 Correct 2 ms 468 KB Output is correct
14 Correct 5 ms 852 KB Output is correct