B 减成一
存在n个数,每次操作可以任选一个区间使得区间内的所有数字减一。问最少多少次操作,可以让所有数都变成1。
数据保证一定有解。
题解
第一种方法,可以理解为分治。
用两个1
确定一个范围,区间内每个数分别减去区间内最小的数字-1
后得到的值。循环该步骤,直到所有数字均为1。
第二种方法,差分数组。
每一次减一的操作对应到差分数组上大概可以表示为,选取一个数减一,并选取一个数加一。加一操作可以在区间外。详见差分数组博客 。
题目要求将所有数变为1,那么对应的差分数组会变成除了第一个元素为1之外,元素全为0的数组。
那么总的操作数即为差分数组中的整数之和减一。
AC代码
第一种
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
vector<int> v, f;
int valid = 0;
int ans = 0;
void ope(int a, int b) {
int t = *min_element(v.begin() + a + 1, v.begin() + b);
t-=1;
for (int i = a + 1; i < b; i++) {
v[i] -= t;
if (v[i] == 1) {
f.push_back(i);
valid++;
}
}
ans += t;
}
int main() {
int T;
scanf("%d", &T);
while (T--) {
ans = 0;
int n;
v.clear();
f.clear();
valid = 0;
scanf("%d", &n);
v.resize(n + 1);
f.push_back(0);
f.push_back(n + 1);
for (int i = 1; i <= n; i++) {
scanf("%d", &v[i]);
if (v[i] == 1) {
f.push_back(i);
valid++;
}
}
while (valid != n) {
sort(f.begin(), f.end());
// 这里可以用set, 会快一点。
vector<int> temp = f;
for (int i = 0; i < temp.size()-1; i++) {
if (temp[i] + 1 != temp[i + 1]) {
ope(temp[i], temp[i + 1]);
}
}
}
printf("%d\n", ans);
}
return 0;
}
第二种
#include <bits/stdc++.h>
//#pragma GCC optimize(2)
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ull unsigned long long
#define ll long long
#define rep(i, x, y) for(int i=x;i<=y;i++)
#define mms(x, n) memset(x, n, sizeof(x))
#define pre(i, x, y) for(int i = x; i >=y; i--)
#define INF (0x3f3f3f3f)
#define mod (1e9+7)
using namespace std;
int main() {
int T;
cin >> T;
while (T--) {
int n;
cin >> n;
int a[n], b[n];
for (auto &i : a) cin >> i;
int sum = b[0] = a[0];
rep(i, 1, n - 1) {
b[i] = a[i] - a[i - 1];
if (b[i] > 0) sum += b[i];
}
cout << sum - 1 << endl;
}
return 0;
}