暴力枚举
其实这种问题答案下手地方就在边界,每个子问题都是由后一个状态推过来的
c[n]必为0, 因为后面没有元素
如果枚举当前元素值大于元素集合后面最大值,c[i] = 0
如果枚举当前元素值等于元素集合后面最大值,c[i] = 1
如果枚举当前元素值小元素集合后面最大值,c[i] = maxv - a[i] + 1
时间复杂度 O(n)
C++ 代码
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 10;
int n;
int a[N];
int c[N];
inline void solve()
{
cin >> n;
for (int i = 1; i <= n; i ++ ) cin >> a[i];
int ans = a[n];
c[n] = 0;
for (int i = n - 1; i >= 1; i -- )
{
if (a[i] > ans) c[i] = 0;
else if (a[i] == ans) c[i] = 1;
else c[i] = ans - a[i] + 1;
if (ans < a[i]) ans = a[i];
}
for (int i = 1; i <= n; i ++ ) cout << c[i] << " ";
}
signed main()
{
cin.tie(nullptr) -> sync_with_stdio(0);
solve();
return 0;
}