Codeforces-703B Mishka and trip

题目大意:

给你n个点,标号为1-n,并且相邻的两个点之间存在路径,其路径权值为路径两端点权值的乘积,现在在1-n中有k个点,这k个点对其它任意一点都需要存在路径,但是两个点之间的路径不能超过一条。也就是没有重边,现在问你整个图的路径权值之和是多少

解题思路:

因为这题的数据为10w所以不可能用两两之间路径来枚举,那么有个简单的思路就是只需要知道与i点相连的路径权值之和就可以了。

而对于k个点来说,直接计算与整个图的点的路径权值之和就行了。

对于第i个点与它相连的只存在两种情况,k中的点是i的邻点或不是

这样先预处理出所有点的权值和,以及k个点的权值和再

代码:

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
const int maxn = 1e5 + 5;
int c[maxn], id[maxn];
int main(){
    int a, b, n, k, sum = 0, sumk = 0;
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; ++i){
        scanf("%d", &c[i]);
        sum += c[i];
    }
    for(int i = 0; i < k; ++i) {
        scanf("%d", &id[i]);
        id[i] -= 1;
        sumk += c[id[i]];
    }

    LL ans = 0, tmp;
    for(int i = 0, j = 0; i < n; ++i){
        a = (i + 1) % n;
        b = (i - 1 + n) % n;
        if(i == id[j]){
            tmp = sum - c[i];
            ++j;
        }else{
            tmp = c[a] + c[b] + sumk;
            int pos = lower_bound(id, id + k, a) - id;
            if(id[pos] == a) tmp -= c[a];
            pos = lower_bound(id, id + k, b) - id;
            if(id[pos] == b) tmp -= c[b];
        }
        ans += tmp * c[i];
    }
    printf("%I64d
", ans >> 1);
    return 0;
}


原文地址:https://www.cnblogs.com/wiklvrain/p/8179409.html