牛客算法周周练15B - 可持久化动态图上树状数组维护01背包(贪心)
题目大意:
给出 n 个数,你的任务是清除这些数,可以从任意位置删除一个数,删除的代价是 i * a[i] 这个数删除后,后面的数统一前移(下标-1),输出删除所有数的最小代价。
解题思路:
题目大水题,最开始以为全是正数,以为只要一直删除第一个元素就可以得到最小代价,然后wa了一次,没有考虑负数,理解题意以后很明显的一道贪心题,想要得到最小代价,遇到负数时就要使下标尽可能大,从后往前考虑,可以保证每一个负数的代价最小化,也就是原来的下标,考虑完负数后考虑正数,正数一直删除第一个元素就可以了,使下标始终是 1 ,综上,遇到负数时乘他原有的下标,遇到正数乘 1 (也就是不变),从1 -> n推过去就可以了。记得开long long ,因为long long 又WA了一次。AC代码:
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N = 1e6+50;
ll a[N];
int main()
{
ll ans = 0;
int n;
cin >> n;
for (int i = 1; i <= n; i ++)
cin >> a[i];
for (int i = 1; i <= n; i ++)
if( a[i] < 0)
ans += a[i] * i;
else
ans += a[i];
cout << ans << endl;
return 0;
}