学而思编程周赛语言普及奠基组 | 2025年春第15周T2 散步-CSDN博客

本文链接：https://blog.csdn.net/guolianggsta/article/details/149027291

欢迎大家订阅我的专栏：算法题解：C++与Python实现！
本专栏旨在帮助大家从基础到进阶，逐步提升编程能力，助力信息学竞赛备战！

专栏特色
1.经典算法练习：根据信息学竞赛大纲，精心挑选经典算法题目，提供清晰的代码实现与详细指导，帮助您夯实算法基础。
2.系统化学习路径：按照算法类别和难度分级，从基础到进阶，循序渐进，帮助您全面提升编程能力与算法思维。

适合人群：

准备参加蓝桥杯、GESP、CSP-J、CSP-S等信息学竞赛的学生
希望系统学习C++/Python编程的初学者
想要提升算法与编程能力的编程爱好者

【题目描述】

小猴解决自己的二进制问题之后，想要出去散步放松一下。

小猴正在散步的时候，突然出现了一只魔物! 幸好小猴随身携带了 $n$ 把刀，小猴可以选取手里的任意两把刀 $i, j (i$ $<$ $j)$ ，对于魔物可以造成 $a_i+a_j)^2$ 点伤害，每把刀只能重复使用 $n - 1$ 次，也就是说小猴可以打出的最大总伤害为：

$\sum\limits_{1\le i\lt j\le n}(a_i+a_j)^2$

小猴知道自己手里的刀一定可以消灭这个魔物，但是小猴想要知道用手里的刀能打出的总伤害最大是多少，由于结果比较大，只需要输出结果除以 $10^9+7$ 的余数。

提示： $a+b)^2=a^2+2ab+b^2$ 。

【输入】

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

第二行，包含 $n$ 个正整数 $a_1,a_2,…,a_n$ 。

【输出】

一行一个整数，表示结果。

【输入样例】

3
1 2 3

【输出样例】

【代码详解】

#include <bits/stdc++.h>
using namespace std;

#define int long long  // 定义长整型别名
const int N = 1000005, mod = 1e9 + 7;  // 定义常量

int n;          // 数组长度
int a[N];       // 存储输入数组
int sa[N];      // 前缀和数组
int sum1, sum2; // sum1: 第一部分和，sum2: 第二部分和

signed main()
{
    // 输入数组长度
    cin >> n;
    
    // 计算前缀和数组
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        sa[i] = sa[i - 1] + a[i];  // 计算前缀和
    }

    // 计算第一部分和：Σ(n-1)*a[i]²
    for (int i = 1; i <= n; i++)
    {
        sum1 = (sum1 + (n - 1) * a[i] * a[i] % mod) % mod;
    }

    // 计算第二部分和：2*Σ(sa[i-1]*a[i])
    for (int i = 1; i <= n; i++)
    {
        sum2 = (sum2 + sa[i - 1] * a[i] % mod) % mod;
    }
    sum2 = sum2 * 2 % mod;  // 乘以2

    // 输出最终结果（两部分和相加取模）
    cout << (sum1 + sum2) % mod << endl;

    return 0;
}

【运行结果】

3
1 2 3
50