MuWuのカードゲーム|接頭辞と接尾辞
oj.cupacm.com
制限時間:1秒
メモリ制限:256MB
問題の説明
MuWuはカードゲームを発明しました。彼はあなたを彼と一緒にプレイするように招待したいと思っています。あなたがとても親切なら、彼を拒否するべきではありません。
ゲームのルールは非常に単純です。テーブルにはn枚のカードがあり、i番目のカードは整数値aiを持ち、MuWuは数字xを指定し、最初のx枚のカードを取る必要があり、残りのカードは任意に選んだ、彼はあなたに尋ねたい、あなたの手札のカードの価値の最大合計は何ですか。
ゲームの1ラウンドが単純すぎるため、MuWuはあなたとqラウンドをプレイしたいと考えています。
入力
最初の行の2つの数字nとqは、ゲームのn枚のカードとqラウンドを表します。
2行目にはn個の数字があり、i番目の数字はi番目のカードの整数値aiを表します。
q行が続き、各行には番号xがあり、最初のxカードを取る必要があり、残りのカードはオプションです。
- 1≤ N ≤2×10 5
- 1≤ Q ≤2×10 5
- 1≤ X ≤ n個
- 1≤ai≤10 4
出力
q行、各行は質問への回答を表します。
サンプル入力
5 5
-1 2 -3 4 -5
1
2
3
4
5
サンプル出力
5
5
2
2
-3
問題は、前のx人の兄弟の数とxの後の正の数の合計を見つけることであることが簡単にわかります。
#include<cstdio>
using namespace std;
int x[200005],pre[200005],rear[200005];//原数组,前面选中的数的和,之后的正数之和
int n,m,sum = 0;
int main(){
scanf("%d %d",&n,&m);
for(int i = 1;i <= n;++i)
scanf("%d",&x[i]),pre[i] = pre[i - 1] + x[i];
for(int i = n;i>= 1;--i)
if(x[i] > 0)
sum += x[i],rear[i] = sum;
else
rear[i] = sum;
while(m--){
scanf("%d",&sum);
printf("%d\n",pre[sum] + rear[sum + 1]);
}
return 0;
}