题目
题意
有 n 堆卡片,每堆都有一定数量卡片和一个惩罚值,在游戏开始前可以把第一堆移到最后面。每搬一堆卡片,会增加和该堆卡片数量相等的分数,但是会减去惩罚值,当分数为负或全搬完则结束比赛。问怎么才能在移动最少堆卡片的情况下搬尽量多的卡片。(惩罚值总数和卡片总数是相等的)
题解
前缀和。因为惩罚值和卡片总数是相同的,所以最多移动 (n-1) 次,就能保证拿走所有卡片。
先假设从第一堆开始拿,如果在拿的过程中,总分变成负数,则把该堆及之前的所有堆按顺序移到末尾。然后从下一堆开始重复此过程,一遍循环后一定能得出答案。
在处理的过程中用上前缀和,就可以把时间复杂度将为 O(n)。
代码
#include <algorithm>
#include <bitset>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <climits>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
using namespace std;
const int MAX = 1000005;
int a[MAX],b[MAX];
int sum[MAX];
int main(){
int n;
while(cin >> n){
for(int i=0;i<n;++i){
scanf("%d",a + i);
}
for(int i=0;i<n;++i){
scanf("%d",b + i);
}
memset(sum,0,sizeof(sum));
sum[0] = a[0] - b[0];
for (int i = 1; i < n; ++i) {
sum[i] = (sum[i-1] + a[i] - b[i]);
}
int pre = -1;
for(int i=0;i<n-1;++i){
if(pre == -1){
if(sum[i] < 0)
pre = i;
}
else if(sum[i] - sum[pre] < 0){
pre = i;
}
}
cout << pre + 1 << endl;
}
return 0;
}