洛谷 P1351 [NOIP 2014 提高组] 联合权值-普及/提高-

最新推荐文章于 2026-06-23 17:20:50 发布

原创最新推荐文章于 2026-06-23 17:20:50 发布 · 1.4k 阅读

13 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#算法 #c++ #洛谷

洛谷专栏收录该内容

199 篇文章

订阅专栏

P1351 [NOIP 2014 提高组] 联合权值

题目背景

NOIP2014 提高组 D1T2

题目描述

无向连通图 $G$ 有 $n$ 个点， $n - 1$ 条边。点从 $1$ 到 $n$ 依次编号,编号为 $i$ 的点的权值为 $W_i$ ，每条边的长度均为 $1$ 。图上两点 $(u, v)$ 的距离定义为 $u$ 点到 $v$ 点的最短距离。对于图 $G$ 上的点对 $(u, v)$ ，若它们的距离为 $2$ ，则它们之间会产生 $Wv×WuW_v \times W_u$ 的联合权值。

请问图 $G$ 上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？

输入格式

第一行包含 $1$ 个整数 $n$ 。

接下来 $n - 1$ 行,每行包含 $2$ 个用空格隔开的正整数 $u, v$ ，表示编号为 $u$ 和编号为 $v$ 的点之间有边相连。

最后 $1$ 行，包含 $n$ 个正整数，每两个正整数之间用一个空格隔开，其中第 $i$ 个整数表示图 $G$ 上编号为 $i$ 的点的权值为 $W_i$ 。

输出格式

输出共 $1$ 行，包含 $2$ 个整数，之间用一个空格隔开,依次为图 $G$ 上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大，输出它时要对 $10007$ 取余。

输入输出样例 #1

输入 #1

输出 #1

20 74

说明/提示

样例解释

本例输入的图如上所示，距离为 $2$ 的有序点对有 $(1, 3)$ 、 $(2, 4)$ 、 $(3, 1)$ 、$(3,5) $、$ (4,2)$ 、$(5,3) $。

其联合权值分别为 $2, 15, 2, 20, 15, 20$ 。其中最大的是 $20$ ，总和为 $74$ 。

数据说明

对于 $30%30\%$ 的数据， $\leq 100$ ；
对于 $60%60\%$ 的数据， $\leq 2000$ ；
对于 $100%100\%$ 的数据， $\leq 2\times 10^5$ ， $W_i \leq 10000$ 。

保证一定存在可产生联合权值的有序点对。

solution

对于一个节点 u ，它的所有相邻顶点之间的距离和为 2，找到其中最大的两个 w[i] 和 w[j]，w[i] * w[j] 为局部最大值。对于要算 $∑i,j,i≠j(w[i]∗w[j])\sum_{i,j,i \neq j }(w[i] * w[j])$ ，由于点太多，不能两两计算，而是求它们的和 S 与平方和 T。则
$S2−T=∑i,j,i≠j(w[i]∗w[j])S^2 - T = \sum_{i,j,i \neq j }(w[i] * w[j])$

代码

#include <iostream>
#include "bit"
#include "vector"
#include "unordered_set"
#include "set"
#include "queue"
#include "algorithm"
#include "bitset"
#include "cstring"

using namespace std;

/*
 * 贪心：在连接前面端点的情况下，下一站尽可能远
 */


const int N = 2e5 + 1, M = 10007;

int n, a[N];
vector<int> e[N];

int main() {
    cin >> n;
    for (int i = 1; i < n; i++) {
        int x, y;
        scanf("%d %d", &x, &y);
        e[x].push_back(y);
        e[y].push_back(x);
    }

    for (int i = 1; i <= n; i++) cin >> a[i];

    int sum = 0, Max = 0;
    for (int i = 1; i <= n; i++) {
        int s = 0, t = 0, _1st = 0, _2nd = 0;
        for (int u: e[i]) {
            s = (s + a[u]) % M;
            if(a[u] > _1st) _2nd = _1st, _1st = a[u];
            else if(a[u] > _2nd) _2nd = a[u];
            t = (t + a[u] * a[u]) % M;
        }
        Max = max(Max, _1st * _2nd);
        s = (s * s - t + M) % M;
        sum = (sum + s) % M;
    }

    cout << Max << " " << sum;
}