#3234. 「POI2019 R1」Pomniejszenie

题目描述

题解

我们可以考虑第一位 $p$ 使得 $a[p]<b[p]$ 的需要满足什么条件，故前面不等的个数和加上第 $p$ 位 $a[p]\ge b[p]$ 的和 $\le k$ ，且加上 $n-p$ ，并且 $a[p]<b[p]$ 且 $a[p]$ 可以更小的话再加上 $1$ ，总和要 $\ge k$ 。于是我们可以找到最大的满足条件的 $p$ ，然后把前面改成相等，后面的不是 $9$ 的改成 $9$ ，如果没到个数的话那就从后往前把原来是 $9$ 的改成 $8$ 即可。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int T,k,n,a[N],b[N],c[N];
char A[N],B[N];
void work(){
	scanf("%s%s%d",A,B,&k);
	n=strlen(A);int p=-1,v=0;
	for (int i=0;i<n;i++) a[i]=A[i]^48,b[i]=B[i]^48;
	for (int u,i=0;i<n;i++){
		if (b[i]){
			u=v+(a[i]>=b[i]);
			if (u<=k && u+n-i-1+(a[i]<b[i]&&b[i]>1)>=k) p=i;
		}
		v+=(a[i]!=b[i]);
	}
	if (p<0){puts("-1");return;}v=0;
	for (int i=0;i<p;i++) v+=(a[i]!=b[i]),a[i]=b[i];
	if (a[p]>=b[p]) a[p]=b[p]-1,v++;
	else if (v<k && a[p]!=b[p]-1) a[p]=b[p]-1,v++;
	for (int i=p+1;i<n;i++)
		if (a[i]<9 && v<k) a[i]=9,v++;
	for (int i=n-1;i>p;i--)
		if (A[i]=='9' && v<k) a[i]=8,v++;
	while(v<k) v++,a[p]--;
	for (int i=0;i<n;i++) putchar(a[i]^48);puts("");
}
int main(){for (scanf("%d",&T);T--;work());return 0;}