[蓝桥杯 2022 省 A] 求和

最新推荐文章于 2024-05-10 19:35:58 发布

我不会画饼鸭

最新推荐文章于 2024-05-10 19:35:58 发布

阅读量88

点赞数 1

分类专栏：蓝桥杯文章标签：蓝桥杯算法职场和发展

本文链接：https://blog.csdn.net/qq_61715584/article/details/129665961

版权

蓝桥杯专栏收录该内容

10 篇文章 0 订阅

订阅专栏

文章目录

[蓝桥杯 2022 省 A] 求和
思路分析
- 如何优化？
demo
总结

[蓝桥杯 2022 省 A] 求和

题目描述

给定 $n$ 个整数 $a_{1}, a_{2}, \cdots, a_{n}$ , 求它们两两相乘再相加的和，即

$S=a_{1} \cdot a_{2}+a_{1} \cdot a_{3}+\cdots+a_{1} \cdot a_{n}+a_{2} \cdot a_{3}+\cdots+a_{n-2} \cdot a_{n-1}+a_{n-2} \cdot a_{n}+a_{n-1} \cdot a_{n}$

输入格式

输入的第一行包含一个整数 $n$ 。

第二行包含 $n$ 个整数 $a_{1}, a_{2}, \cdots a_{n}$ 。

输出格式

输出一个整数 $S$ ，表示所求的和。请使用合适的数据类型进行运算。

样例 #1

样例输入 #1

4
1 3 6 9

样例输出 #1

提示

对于 $\%$ 的数据, $\leq n \leq 1000,1 \leq a_{i} \leq 100$ 。

对于所有评测用例, $\leq n \leq 2\times10^5,1 \leq a_{i} \leq 1000$ 。

蓝桥杯 2022 省赛 A 组 C 题。

思路分析

刚看到这题，想到的是直接求和就ok，代码如下（这里附加一点，由于懒得看数据大小，直接开long long）

#include <iostream>
using namespace std;

const int N = 2e5 + 10;
long long sum;
long long q[N];
int main()
{
    int n; 
    cin>>n;

    for(int i = 0; i < n; i++) cin>>q[i];
    
    for(int i = 0; i < n; i++)
    {
        long long t = 0;
        for(int j = i + 1; j < n; j++)
            t += q[i] * q[j];
        sum += t;
    }
    cout<<sum;
    return 0;
}

结果就是30分，其他的都超时了

如何优化？

仔细观察，其实我们会发现一个东西，每一项的形式如下

sum[i] = q[i] * (q[2] + ... + q[n])

我们要求的是 sum[1] + …+ sum[n]
在我们第一份求sum中每一项的时候，时间复杂度是 O（N ^ 2）,由于上面我们发现的这个式子，其实求出sum中每一项的过程我们可以优化到 O(N),这样就可以完美的解决超时的问题了

demo

#include <iostream>
using namespace std;

const int N = 2e5 + 10;
long long sum[N];
long long q[N];
long long res = 0;
int main()
{
    int n; 
    cin>>n;

    for(int i = 1; i <= n; i++)
    {
        cin>>q[i];
        sum[i] = sum[i - 1] + q[i];
    }
    
    for(int i = 1; i <= n; i++)
    {
        sum[i] = sum[n] - sum[i];
        sum[i] = sum[i] * q[i];
        res += sum[i];
    }
    
    cout<<res<<endl;
    return 0;
}