Codeforces Round #365 (Div. 2) B - Mishka and trip

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+;

 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=; summ=;
         for(int i=;i<=n;i++)
         {
             scanf("%d",&a[i]);
             sum+=a[i];    //所有点的和
             vis[i]=;
         }
         for(int i=;i<=k;i++)
         {
             scanf("%d",&b[i]);
             summ+=a[b[i]];  //城市的和
             vis[b[i]]=;
         }

         LL ans=;
         for(int i=;i<=k;i++)
         {
             ans+=a[b[i]]*(sum-a[b[i]]);
         }

         //去重
         LL temp=;
         for(int i=;i<=k;i++)
         {
             temp+=a[b[i]]*(summ-a[b[i]]);
         }
         temp/=;
         ans-=temp;

         //计算环上还未计算的
         for(int i=;i<n;i++)
         {
             if(vis[i]== && vis[i+]==)
                 ans+=a[i]*a[i+];
         }
         if(vis[]== && vis[n]==)
             ans+=a[]*a[n];
         printf("%lld\n",ans);
     }
     return ;
 }