Codeforces 509C. Sums of Digits 贪心 + 模拟

题意

有一个数组a, 它是严格递增的，给你数组b，b[i] = a[i]按位加和

问你要保证a
最小的，数组a是什么样的

思路

贪心，a
最小应有a[n-1]也是最小的

我用一个vector保存a[i]的各位数字

每次迭代时，大情况分三类就是b[i] > b[i-1] 且用之前的位数保存不下b[i]，那么直接重构这个vector即可

如果只是b[i] > b[i-1]，那么从低位到高位让每位最多增加到9，直至总共增加了b[i] - b[i-1]

除上述两种情况外，基本做法是直接给vector加前导一位0，然后从低位到高位让每位最多减少到0，直至总共减少了b[i] - b[i-1] +1，然后让再高一位加1

但是这里，有个很麻烦的事情，就是再高一位是9。。。

这个情况特别碎，前面的构造方法就不对了。。。这里还要分两种情况考虑。。。具体的请看代码吧，说不太清了。。。

实现

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
vector<int> ans , pre;
int b[302];
int n;
void init(int b){
ans.clear();
while (b > 0){
if (b >= 9){
ans.push_back(9);
b -= 9;
}
else{
ans.push_back(b);
b = 0;
}
}
}
void print(vector<int>& ans){
int flag = 0;
for (int i=ans.size()-1;i>=0;i--){
if (ans[i] == 0 && flag == 0)
continue;
printf("%d",ans[i]);
flag = 1;
}
puts("");
}

int main(){
cin>>n;
for (int i=0;i<n;i++){
scanf("%d",&b[i]);
}
for (int i=0;i<n;i++){
pre.clear();
pre.insert(pre.end(),  ans.begin(),  ans.end()  );

int man = ans.size() * 9;
if (b[i] > man){
init(b[i]);

print(ans);
continue;
}
if (b[i] > b[i-1]){
int cha = b[i] - b[i-1];
for (int i=0;i<ans.size()&&cha>0;i++){
if (9-ans[i] <= cha){
cha -= 9-ans[i];
ans[i] = 9;
}
else{
ans[i] += cha;
cha = 0;
}
}
}
else{
int cha = b[i-1] - b[i] + 1;
ans.push_back(0);
int k;
vector<int> tmp(ans);
int pp = 0;
for (k=0;k<ans.size()&&cha>0;k++){
if (ans[k] <= cha){
cha -= ans[k];
ans[k] = 0;
}
else{
ans[k] -= cha;
pp = ans[k];
cha = 0;
}
}
if (ans[k] != 9){
ans[k] ++;
swap(ans[0],ans[k-1]);

}
else{
cha = b[i-1] - b[i] + 1;
ans.clear();
ans.insert(ans.end(),  tmp.begin(),  tmp.end()  );
int id = k;
for (;ans[k]==9;k++)
;
ans[k]++;
int sum;
if (id*9-pp > (k-id)*9){
for (int j=0;j<(k-id);j++){
ans[j] = 9;
}
ans[k-id] = pp;
for (int j=k-id+1;j<k;j++){
ans[j] = 0;
}
print(ans);
continue;
}
sum = id * 9 - pp;
int j;
for (j=k-1;j>=0&&sum>0;j--){
if (ans[j] <= sum){
sum -= ans[j];
ans[j] = 0;
}
else{
ans[j] -= sum;
sum = 0;
}
}
for (;j>=0;j--){
ans[j] = 9;
}
}
}
print(ans);
}

return 0;
}