CodeForces 703B Mishka and trip

简单题。

先把环上的贡献都计算好。然后再计算每一个$capital$ $city$额外做出的贡献值。

假设$A$城市为$capital$ $city$，那么$A$城市做出的额外贡献：$A$城市左边城市$L$和右边城市$R$都不能和$A$做出贡献，之前存在的$capital$ $city$和$A$城市不能做出贡献，要注意的是$L$和$R$中存在$capital$ $city$的情况。剩下的城市都可以和$A$做出贡献。

#pragma comment(linker, "/STACK:1024000000,1024000000")

#include<cstdio>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<stack>

#include<iostream>

using namespace std;

typedef long long LL;

const double pi=acos(-1.0),eps=1e-;

void File()

{

    freopen("D:\\in.txt","r",stdin);

    freopen("D:\\out.txt","w",stdout);

}

const int maxn=;

int n,k;

LL c[maxn],sum,y,p;

bool f[maxn];

int main()

{

    scanf("%d%d",&n,&k);

    for(int i=;i<=n;i++) scanf("%lld",&c[i]);

    LL ans=; c[n+]=c[];

    for(int i=;i<=n;i++)  ans=ans+c[i]*c[i+],y=y+c[i];

    for(int i=;i<=k;i++)

    {

        int x; scanf("%d",&x); sum=y;

        int L=x-,R=x+; if(L==) L=n; if(R==n+) R=;

        sum=sum-c[L]-c[R]-c[x]; sum=sum-p;

        if(f[L]) sum=sum+c[L]; if(f[R]) sum=sum+c[R];

        ans=ans+c[x]*sum; f[x]=; p=p+c[x];

    }

    printf("%lld\n",ans);

    return ;

}