0
点赞
收藏
分享

微信扫一扫

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

心存浪漫 2022-08-30 阅读 54


题目链接:http://codeforces.com/contest/703/problem/B

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

Codeforces Round #365 (Div. 2) Mishka and trip_i++_02

Codeforces Round #365 (Div. 2) Mishka and trip_i++_03

题意:给出我们每一个城市的美丽度,所有城市连成一个环,再给出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);
}



举报

相关推荐

0 条评论