1687: 陣列操作
Time Limit: 1 Sec Memory Limit: 128 MB
[Submit][Status][Web Board]
Description

給你一個初始的長度為n的陣列。（1<=n<=105）

有兩個操作：

Op1(l, r):給兩個整數l和r（1<=l<=r<=當前陣列長度）。你需要計算陣列從l到r的所有元素的和。

Op2(x):給你一個整數x（|x| <= 109），你需要將x新增到陣列的頭部。原先的第一個元素變成第二個元素，第二個元素變成第三個，以此類推。並且陣列的長度增加1.

Input

多組測試資料，處理到檔案結尾。

每組測試資料，首先給出一個N(1 <= N <= 105)， 表示初始的陣列的元素個數。

第二行N個數字表示初始陣列的n個元素，a1 a2 … aN.(|ai| <= 109)。

第三行有一個Q，表示有Q個操作。1 <= Q <= 105

接下來Q行，每行的第一個數表示操作的型別，只可能是1或者2,形式如下：

1 l r：操作1,求l到r的元素的和。

2 x：操作2,將x新增到陣列頭部。

Output

對於每組測試資料，首先輸出"Case x:",表示當前是第x組測試資料。

然後對於每個操作1輸出對應的答案，對於操作2不需要任何輸出。

Sample Input

10
1 2 3 4 5 6 7 8 9 10
4
1 1 10
1 1 1
1 10 10
1 2 7
5
6 7 8 9 10
9
2 5
2 4
1 2 7
2 3
2 2
2 1
1 1 10
1 1 1
1 10 10

Sample Output

Case 1:
55
1
10
27
Case 2:
45
55
1
10

HINT

Source
AC程式碼~：

#include <stdio.h>
#include <string.h>
long long a[200005],b[200005];
int main()
{
    int n,c = 0;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i = n; i > 0; i--)//陣列indext倒過來存，加一個數時就方便了
        {
            scanf("%lld",&a[i]);
        }
        for(int j = 1; j <= n; j++)//最後一個數到倒數第j個數的和
        {
            b[j] = b[j-1] + a[j];
        }
        int Q,num,t;
        scanf("%d",&Q);
        printf("Case %d:\n",++c);
        while(Q--)
        {
            scanf("%d",&num);
            if(num==1)
            {
                int l,r;
                scanf("%d%d",&l,&r);
                printf("%lld\n",b[n-l+1]-b[n-r]);//草稿紙上推出這個公式
            }
            else
            {
                n++;
                scanf("%lld",&a[n]);
                b[n] = b[n-1] + a[n];
            }
        }
    }
    return 0;
}
 

(用動態陣列+數狀陣列）時間超限程式碼：
//時間超限，樹狀陣列中整合陣列c更新慢（每次加一個數要更新，耗時啊！！）

#include <stdio.h>
#include <vector>
using namespace std;
vector<long long>a(200005),c(200005);
int n;
int lowbit(int x)
{
    return x&-x;
}
long long SUM(int y)
{
    long long sum = 0;
    for(int i = y; i > 0; i-=lowbit(i))
        sum += c[i];
    return sum;
}
void update(int y,int k)
{
    for(int i = y; i <= n; i+=lowbit(i))
        c[i] += k;
}
int main()
{
    int Q,C = 0;
    while(~scanf("%d",&n))
    {
        for(int i = 1; i<= n; i++)
        {
            scanf("%lld",&a[i]);
            update(i,a[i]);
        }
        scanf("%d",&Q);
        printf("Case %d:\n",++C);
        while(Q--)
        {
            int num,l,r,t;
            scanf("%d",&num);
            if(num==1)
            {
                scanf("%d%d",&l,&r);
                long long sum = SUM(r)-SUM(l-1);
                printf("%lld\n",sum);
            }
            else
            {
                scanf("%d",&t);
                a.insert(a.begin()+1,t);
                n++;
                fill(c.begin(),c.begin()+n+1,0);//動態陣列賦值
                for(int i = 1; i<= n; i++)
                {
                    update(i,a[i]);
                }
            }
        }
    }
    return 0;
}