Codeforces 703B Mishka and trip

Posted 声声醉如兰

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Codeforces 703B Mishka and trip相关的知识,希望对你有一定的参考价值。

题目大意:有n个城市,k个省会,每个城市有一个魅力值,两个城市之间的连线的价值等于两个城市魅力值的乘积。省会城市和每个城市相连,城市i和城市i+1相连,城市n与城市1相连。求所有连线的和。

题目思路:v[k]*v[1]+v[k]*v[2]+……v[k]*v[n]=v[k]*(v[1]+v[2]+……v[n])。为了节省时间预处理城市魅力值的和。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<stdlib.h>
#include<queue>
#include<math.h>
#include<map>
#define INF 0x3f3f3f3f
#define MAX 1000005
#define Temp 1000000000

using namespace std;

int v[MAX],vis[MAX],a[MAX],n,k,len;

int main()
{
    int num,i,j;
    long long sum1,sum2,ans;
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        sum1=0;
        sum2=0;
        ans=0;
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&v[i]);
            sum1+=v[i];//sum1记录城市魅力值的和
        }
        for(i=1;i<=k;i++)
        {
            scanf("%d",&a[i]);
            sum2+=(v[a[i]]);//sum2记录未处理的省会城市魅力值的和
        }

        for(i=1;i<=k;i++)
        {
            num=a[i];
            ans+=(sum1-v[num])*v[num];//算出该省会城市与其他城市连线的和
            ans-=(sum2-v[num])*v[num];//减去重复计算的值(省会之间的连线)
            sum2-=v[num];
            vis[num]=1;
        }

        for(int i=1;i<=n;i++)//计算非省会城市之间的连线
        {
            if(!vis[i]&&!vis[i+1])
                ans+=(v[i]*v[i+1]);
        }
        if(!vis[1]&&!vis[n])
            ans+=(v[1]*v[n]);
        printf("%lld\\n",ans);
    }
    return 0;
}
View Code

 

以上是关于Codeforces 703B Mishka and trip的主要内容,如果未能解决你的问题,请参考以下文章

Educational Codeforces Round 56 Editorial

[树状数组]Mishka and Interesting sum(codeforces703D)

codeforces 703E Mishka and Divisors

cf--703--A-- Mishka and Game

Codeforces Round #365 (Div. 2) D. Mishka and Interesting sum 树状数组

Codeforces Round #365 (Div. 2) D - Mishka and Interesting sum(离线树状数组)