当前位置: 代码迷 >> 综合 >> Codeforces Round #365 (Div. 2) Mishka and trip
  详细解决方案

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

热度:55   发布时间:2023-12-06 03:13:44.0

题目链接: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);
}


  相关解决方案