思路:
就是我们每次维护好下标就可以了,然后开一个vector来纯下每天的值日人编号一直到要求的天数。最后res数组的末尾就是答案。
代码:
#include<iostream>
#include<vector>
using namespace std;
int n,m;
int main(){
cin>>n>>m;
vector<vector<int>> a(n + 1,vector<int> (n + 1));
for(int i = 1;i <= n; i++)
for(int j = 1; j <= n; j++) cin>>a[i][j];
vector<int> res;
res.push_back(1);
res.push_back(2);
while(res.size() <= m){
int first = 1,second = 2;
int tmp = a[first][second];
res.push_back(tmp);
first = second,second = tmp;
}
cout<<res.back();
return 0;
}
评论区