PAT乙级1049 数列的片段和测试点二、三（C++）

松堂飞萤

已于 2022-09-12 11:54:34 修改

阅读量438

点赞数

分类专栏： PAT乙级题目文章标签： c++ 程序设计经验分享

于 2022-09-11 20:15:00 首次发布

本文链接：https://blog.csdn.net/weixin_49119476/article/details/126808520

版权

PAT乙级题目专栏收录该内容

4 篇文章 0 订阅

订阅专栏

给定一个正数数列，我们可以从中截取任意的连续的几个数，称为片段。例如，给定数列 { 0.1, 0.2, 0.3, 0.4 }，我们有 (0.1) (0.1, 0.2) (0.1, 0.2, 0.3) (0.1, 0.2, 0.3, 0.4) (0.2) (0.2, 0.3) (0.2, 0.3, 0.4) (0.3) (0.3, 0.4) (0.4) 这 10 个片段。

给定正整数数列，求出全部片段包含的所有的数之和。如本例中 10 个片段总和是 0.1 + 0.3 + 0.6 + 1.0 + 0.2 + 0.5 + 0.9 + 0.3 + 0.7 + 0.4 = 5.0。

原文链接：点我跳转

输入格式：

输入第一行给出一个不超过 105 的正整数 N，表示数列中数的个数，第二行给出 N 个不超过 1.0 的正数，是数列中的数，其间以一个空格分隔。

输出格式：

在一行中输出该序列所有片段包含的数之和，精确到小数点后 2 位。

输入样例：

4
0.1 0.2 0.3 0.4

输出样例：

5.00

问题分析：

首先考虑数出有多少个片段，并对每个片段求和。但是考虑到片段数量N最大可达到 $10^{5}$ ，如果采用寻找并计算每一个片段的方法做的话复杂度将达到 $O\left ( N^{2} \right )$ ，这个复杂度显然是我们不能接受的。

继续考虑，发现题目要求的片段是连续的。考虑片段的两头，对于第i个数字来说，它左边有i种选择，右边有（n-i+1）种选择，所以第i个数出现次数为 $i\cdot (n-i+1)$ 次。

但是此时，笔者仍过不了测试点二和测试点三。后来经过分析和上网查找，发现了错误原因。

对于测试点二，错误原因是最后存储结果的数开double的话会导致在计算中损失精度而出现结果错误，因此对于保存结果的数字需要开long double。

long double res =0;

对于测试点三，错误原因是，如果选择先计算出现次数并用int存储，该数最高可达到 $2.5\cdot 10^{9}$ 级别，超出了int的表示范围。应该先对片段数相乘。或者用long long存储次数

/*
错误方法
int cishu = i*(n-i+1);
res += cishu* arr[i];
*/

//方法一
res+= arr[i]*i*(n-i+1);
//方法二
#define ll long long
ll cishu = 1ll*i*(n-i+1);
res += cishu*arr[i];

附上完整满分代码：

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

int main()
{
    int n;
    cin>>n;
    vector<double> arr(n+1);
    long double res =0;
    for(int i=1;i<=n;i++)
    {
        cin>>arr[i];
        ll cishu = 1ll*i*(n-i+1);
        res += arr[i]*cishu;
    }
    printf("%0.2Lf",res);
    return 0;
}