1. 程式人生 > >PAT乙級—1049. 數列的片段和(20)-native

PAT乙級—1049. 數列的片段和(20)-native

給定一個正數數列,我們可以從中擷取任意的連續的幾個數,稱為片段。例如,給定數列{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

思路:首先這道題,迴圈肯定是會超時的,既然它是連續數列,那就數學公式吧,找出其中的規律,我們計算它們的和就是計算每個數出現的次數,如 1+2 1+2+3 2+3 2+3+4 3+4 1+2+3+4+5 1+2+3+4+5
他們出現的次數有這樣的規律:
1:因為是連續所以有1 12 123 1234有四次
2:2 23 234 左邊還有個1,還可以在12,123,1234裡
3: 3 34 左邊還有個2,有23 234,還有個1,有123 1234


4:4 左邊有個3,所以有34,還有個2,所以有234,還有個1,所以有1234
看到這裡大家明白了嗎,1右邊有3個加上1本身是一共四個,2右邊2個加上本身一共三個但左邊有一個,還能組成新三個,所以乘以3*2,3右邊有1個加上本身一共兩個,但左邊有兩個,每一個都能都能構成新2個,所以2*3,4沒有加上本身有一個,但左邊有三個,每一個都能夠成新1個所以4*1。
所以最後公式就是:sum+=左邊的個數+1*右邊的個數+1*a[i];也就是sum+=(i+1)a[i](N-i);
注:浮點數相乘的順序可能會影響精度,你也可以這樣寫sum+=(double)(N-i)*(double)(i+1)*a[i];

#include<iostream>
#include<cstdio>
using namespace std;
int main(){
    int N;
    cin>>N;
    double a[N],sum=0;
    for(int i=0;i<N;i++){
        scanf("%lf",&a[i]);
        sum+=(i+1)*a[i]*(N-i);
    }
    printf("%.2lf", sum);
    return 0;
}

題目連結: