题目链接:http://codeforces.com/contest/703/problem/B
题意:给出我们每一个城市的美丽度,所有城市连成一个环,再给出k个首都,然后如果某一个城市是首都,那么它会和所有的其他城市都练一条边,边权为边两边的城市的美丽度的积,没有重边的情况下,问我们总的边权和为多少。
这题应该没有什么特定的方法,小编这里的做法是记录每一个城市和它相连的边的权值和再除以2.
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int maxn = 100000+5;
int c[maxn], cp[maxn];
int main()
{int n,k,sum1 = 0, sum2 = 0;scanf("%d%d", &n, &k);for(int i=0; i<n; i++) scanf("%d", &c[i]),sum1 += c[i];for(int i=0; i<k; i++) scanf("%d", &cp[i]),cp[i]--,sum2 += c[cp[i]];LL ans = 0, sum;sort(cp,cp+k);for(int i=0,j=0; i<n; i++){int a = (i+1)%n;int b = (i-1+n)%n;if(i == cp[j]){sum = sum1-c[i];j++;}else{sum = c[a] + c[b] + sum2;int pos = lower_bound(cp, cp+k, a)-cp;if(cp[pos] == a) sum -= c[a];pos = lower_bound(cp, cp+k, b)-cp;if(cp[pos] == b) sum -= c[b];}ans += sum * c[i];}printf("%I64d\n", ans/2);
}