C. Long Beautiful Integer (构造)

题目

思路：先构造成最小的情况，从前k个数的每个数开始，之后每隔k个数令其等于前面的该数，如 lst[3]=2 i=3(k=4) lst[7] lst[11] lst[15]…=lst[3]=2， 如果此时构造出来的数满足大于原数，则输出。如果不满足则把lst[k] lst[2k] lst[3k]…全体+1，但可能会发生+1后变为10，所以需要从lst[k]往前找到第一个小于9的数位置噢p，将其每隔k个全变为lst[p]+1 lst[p+k]=lst[p]+1…然后再将p—k的数每隔k个全部转化为0，因为已经保证新数一定大于原数了，将后面的变为最小。

Code:

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int Max = 1e6 + 5;
int a[Max], b[Max];

int main()
{
    
    
	int n, k;cin >> n >> k;
	for (int i = 1;i <= n;i++)
	{
    
    
		char p;cin >> p;
		a[i] = p - '0';
	}
	for (int i = 1;i <= k;i++)
	{
    
    
		int g = i;
		while (g <= n)
		{
    
    
			b[g] = a[i];
			g += k;
		}
	}
	int f = 1;
	for (int i = 1;i <= n;i++)if (a[i] != b[i]) {
    
     if(b[i]<a[i]) f = 0;break; }
	if (f == 0)
	{
    
    
		int g, q;
		for (int i = k;i >= 1;i--)
		{
    
    
			if (a[i] <= 8)
			{
    
    
				q = i;g = i;break;
			}
		}
		while (g <= n)
		{
    
    
			b[g] = a[q] + 1;
			g += k;
		}
		for (int i = q+1;i <= k;i++)
		{
    
    
			int g = i;
			while (g <= n)
			{
    
    
				b[g] = 0;
				g += k;
			}
		}
	}
	cout << n << endl;
	for (int i = 1;i <= n;i++)cout << b[i];
}