Codeforces Round #365 (Div. 2) B

http://codeforces.com/contest/703/problem/B

题意：

每个点都有一个值，每条边的权值为这两个点相乘。1~n成环。现在有k个城市，城市与其他所有点都相连，计算出所有边权值和。

思路：

每个城市要与其它所有城市都相连，所以我们可以先去计算城市的，先预处理一下，计算出所有点的值的和，那么每个城市乘以所有点的值的和减去它自身的值。

但是这样的话每两个城市之间就计算了两遍，要减去，具体看代码。

最后再考虑环上的点，如果还有边未计算，就加上去。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
typedef long long LL;

const int maxn=1e5+5;

int n,k;
int a[maxn];
int b[maxn];
int vis[maxn];
LL sum;
LL summ;

int main()
{
    //freopen("D:\input.txt","r",stdin);
    while(~scanf("%d%d",&n,&k))
    {
        sum=0; summ=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];    //所有点的和
            vis[i]=0;
        }
        for(int i=1;i<=k;i++)
        {
            scanf("%d",&b[i]);
            summ+=a[b[i]];  //城市的和
            vis[b[i]]=1;
        }

        LL ans=0;
        for(int i=1;i<=k;i++)
        {
            ans+=a[b[i]]*(sum-a[b[i]]);
        }
        
        //去重
        LL temp=0;
        for(int i=1;i<=k;i++)
        {
            temp+=a[b[i]]*(summ-a[b[i]]);
        }
        temp/=2;
        ans-=temp;
        
        //计算环上还未计算的
        for(int i=1;i<n;i++)
        {
            if(vis[i]==0 && vis[i+1]==0)
                ans+=a[i]*a[i+1];
        }
        if(vis[1]==0 && vis[n]==0)
            ans+=a[1]*a[n];
        printf("%lld
",ans);
    }
    return 0;
}