可爱萌新在线求助
查看原帖
可爱萌新在线求助
735763
_ChongYun_楼主2023/10/17 17:18
#include<bits/stdc++.h>
using namespace std;
int n,k,ans=0,sum=0,now=0;
int val[114514],star[114514];
int main(){
	cin>>n>>k;
	for(int i=1;i<=n;i++) cin>>val[i],sum+=val[i];
	for(int i=1;i<=k;i++) cin>>star[i];
	for(int i=1;i<=n;i++) ans+=val[i]*val[((i!=n)?i+1:1)];
	for(int i=1;i<=k;i++) now+=val[star[i]],ans+=val[star[i]]*(sum-now-val[(star[i]!=1)?star[i]-1:n]-val[((star[i]!=n)?star[i]+1:1)]);
	cout<<ans<<endl;
	return 0;
}
2023/10/17 17:18
加载中...