1968D-Permutation Game

题目链接：Permutation Game

从这个p数组为排列数我们可知s和p这两个人一定是在环中进行，所以在最初的min（n,k）回合里都是最优步数，也就意味着棋手决定移动到最终的位置一定是最优解，那么中间的步数都为1，只有最后停留在的地方为剩余的步数*a[i]。

因此我们可以一步步枚举，假设当前步数为最佳步数，那么前面每个点只停留一步，最后一个点将剩余的步数*a[i]，最后比较总和的最大值。定义一个sum，表示在前 𝑖 个回合中出现的位置的数值总和。

代码附上：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e5+5;
int a[N],p[N];
int n,k,pb,ps;

void solve(){
	cin>>n>>k>>pb>>ps;
	for(int i=1;i<=n;i++)cin>>p[i];
	for(int i=1;i<=n;i++)cin>>a[i];

	int max1=a[pb]*k;
	int c=k;
	int sum=a[pb];
	int x=p[pb];
	while(x!=pb&&c>0){
		c--;
		max1=max(max1,sum+a[x]*c);
		sum+=a[x];
		x=p[x];
	}

	int max2=a[ps]*k;
	c=k;
	sum=a[ps];
	x=p[ps];
	while(x!=ps&&c>0){
		c--;
		max2=max(max2,sum+a[x]*c);
		sum+=a[x];
		x=p[x];
	}

	if(max1>max2)cout<<"Bodya"<<"\n";
	else if(max1<max2)cout<<"Sasha"<<"\n";
	else cout<<"Draw"<<"\n";

}

signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int t;cin>>t;
	while(t--){
		solve();
	}

	return 0;
}