B - 可持久化动态图上树状数组维护01背包(思维) (牛客算法周周练15)

传送门

题意： 对于一个数组a，删除任意位置的元素a[i]需要花费a[i]*i的精力，然后i以后的元素都前进一位(即下标-1)。求把整个序列删完的最小代价。

思路：

• 显然对于正数我们都可以将其放在位置1去删除，花费的精力就是a[1]。
• 而对于负数我们可以就在原位置删除，这样得到的总精力会最小。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 1e6 + 5;

int n, sum, a[N];

signed main()
{
    IOS;
    
    cin >> n;
	for(int i = 1; i <= n; i ++){
		cin >> a[i];
		if(a[i] <= 0) sum += a[i] * i;
		else sum += a[i];
	}
	cout << sum << endl;

    return 0;
}