CF95B Lucky Numbers [毒瘤分类讨论贪心/DFS]

19 March 2019 | duanyll | Tags: OI 题解 贪心 思维题

其实代码用不着这么多goto,不过写着方便~

详细分析有时间再写吧

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;

const int MAXN = 1e5+10;

char a[MAXN],ans[MAXN];
int n;

int main(){
	cin >> (a+1);
	int n = strlen(a+1);
	if(n & 1){
ODD_N:
		for(int i = 1;i<=n/2 + 1;i++){
			putchar('4');
		}
		for(int i = 1;i<=n/2 + 1;i++){
			putchar('7');
		}
		putchar('\n');
	}else{	
		int lead4 = 0,lead7 = 0,now = 1;
		while(now <= n){
			if(a[now] < '4'){
				break;
			}else if(a[now] == '4' && lead4 < n/2){
				ans[now] = '4';
				lead4++;
				now++;
			}else if(a[now] < '7' && lead7 < n/2){
				ans[now] = '7';
				lead7++;
				break;
			}else if(a[now] == '7' && lead7 < n/2){
				ans[now] = '7';
				lead7++;
				now++;
			}else{
				goto TRY_LAST_USE_7;
			}
		} 
		if(false){
TRY_LAST_USE_7:
			while(--now){
				if(ans[now] == '4'){
					lead4--;
				}else{
					lead7--;
				}
				if(ans[now] < '7' && lead7 < n/2){
					ans[now] = '7';
					lead7++;
				 	goto OUTPUT;
				}
			}
			goto ODD_N;
		}
OUTPUT:
		for(int i = 1;i<=lead4 + lead7;i++){
			putchar(ans[i]);
		}
		for(int i = 1;i<=n/2 - lead4;i++){
			putchar('4');
		}
		for(int i = 1;i<=n/2 - lead7;i++){
			putchar('7');
		}
		putchar('\n');
	}
}