把数字1...n分别填在n张牌上,背面朝上洗牌打乱顺序,排成一列,游戏过程是先随便猜一个数k,把第k张牌翻开,显示的数字为x,则接着翻开第x张牌,如此循环翻下去,直到要翻开的那个位置的牌之前已经被翻开过为止。统计一共翻开的牌上的数字的和作为得分。
输入数据共2行,第一行两个数,第一个为n,n<=1000,第2个数为k,即首先要翻开的位置。第二行n个整数,分别空格隔开。
样例
输入:
10 3
5 3 8 6 9 4 1 7 2 10
输出:
35 //说明:翻开的牌:8+7+1+5+9+2+3=35
#include<iostream>
using namespace std;
int main(){
int n,k,i,x,y,s,a[1001]={0};
cin>>n>>k;
for(i=1;i<=n;i++)cin>>a[i];
s=0;
x=k;
do{
s+=a[x];
y=x;
x=a[x];
a[y]=0;
}while(a[x]!=0);
cout<<s;
return 0;
}