HDU-6609 Find the answer 题解

题目大意

给你长度为n的一个数组，现在对于数组上的每一个位置i，每一次操作，你可以将1到i-1的这些元素任意一个变成0，使得从第一个元素到第i个元素的总和<=m,对于每个位置i，求出满足上述条件的最小操作数。

解题思路

这题好像就是一个裸的权值线段树。

这边贴一下大佬的博客。

维护区间个数和区间和。

按照数组的大小，从小到大来维护。

完整代码

#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <queue>
#include <deque>
#include <cstring>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <list>
#include <string>
#include <vector>
#include <algorithm>
#include <sstream>
#include <unordered_map>
using namespace std;
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=a;i>=n;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define pll pair<ll,ll>
#define pii pair<int,int>
#define bg begin
#define rbg rbegin
#define ed end
#define endl '\n'
#define dbg(x) cout << #x << "===" << x << endl
typedef double db;
typedef long long ll;
typedef unsigned long long ull;

const int N = 2e5 + 5;
ll a[N], b[N];
ll trn[N << 2], trs[N << 2];
ll ans[N];

void build(int l, int r, int rt) {
	trn[rt] = trs[rt] = 0;
	if (l == r) return;
	int mid = l + r >> 1;
	build(l, mid, rt << 1);
	build(mid + 1, r, rt << 1 | 1);
}

ll query(int l, int r, int rt, ll sum) {
	if (sum <= 0)	return 0;
	if (l == r)		return ceil(1.0 * sum / b[l]);
	int mid = l + r >> 1;
	if (trs[rt << 1 | 1] >= sum)	return query(mid + 1, r, rt << 1 | 1, sum);
	return query(l, mid, rt << 1, sum - trs[rt << 1 | 1]) + trn[rt << 1 | 1];
}

void update(int l, int r, int rt, int x) {
	if (l == r) {
		trs[rt] += b[l];
		trn[rt]++;
		return;
	}
	int mid = l + r >> 1;
	if (x <= mid)	update(l, mid, rt << 1, x);
	else update(mid + 1, r, rt << 1 | 1, x);
	trs[rt] = trs[rt << 1] + trs[rt << 1 | 1];
	trn[rt] = trn[rt << 1] + trn[rt << 1 | 1];
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	//freopen("D:\\测试用\\in.txt", "r", stdin);
	//freopen("D:\\测试用\\out.txt", "w+", stdout);
	int t; cin >> t;
	while (t--) {
		build(1, N, 1);
		ll n, m; cin >> n >> m;
		rep(i, 1, n)	cin >> a[i], b[i] = a[i];
		sort(b + 1, b + 1 + n);
		int len = unique(b + 1, b + 1 + n) - b - 1;
		ll sum = 0;
		rep(i, 1, n) {
			sum += a[i];
			ans[i] = query(1, len, 1, sum - m);
			int pos = lower_bound(b + 1, b + 1 + len, a[i]) - b;
			update(1, len, 1, pos);
		}
		rep(i, 1, n)	cout << ans[i] << ' ';
		cout << endl;
	}
	return 0;
}