1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
| from sys import setrecursionlimit, stdin, stdout, exit from collections import Counter, deque, defaultdict from heapq import heapify, heappop, heappush, nlargest, nsmallest from bisect import bisect_left, bisect_right from datetime import datetime, timedelta from string import ascii_lowercase, ascii_uppercase from math import log, gcd, sqrt, fabs, ceil, floor from types import GeneratorType from typing import TypeVar, List, Dict, Any, Callable
class SA: def __init__(self, x, y): self.x = x self.y = y
def __lt__(self, other): return self.x < other.x
N = int(2e5 + 10) M = 20 INF = int(2e9)
setrecursionlimit(INF)
input = lambda: stdin.readline().rstrip("\r\n") read = lambda: map(int, input().split()) read_list = lambda: list(map(int, input().split()))
class std:
@staticmethod def bootstrap(f, stack=None): if stack is None: stack = []
def wrappedfunc(*args, **kwargs): if stack: return f(*args, **kwargs) else: to = f(*args, **kwargs) while True: if isinstance(to, GeneratorType): stack.append(to) to = next(to) else: stack.pop() if not stack: break to = stack[-1].send(to) return to
return wrappedfunc
letter_to_num = staticmethod(lambda x: ord(x.upper()) - 65) num_to_letter = staticmethod(lambda x: ascii_uppercase[x]) array = staticmethod(lambda x=0, size=N: [x] * size) array2d = staticmethod( lambda x=0, rows=N, cols=M: [std.array(x, cols) for _ in range(rows)]) max = staticmethod(lambda a, b: a if a > b else b) min = staticmethod(lambda a, b: a if a < b else b) filter = staticmethod(lambda func, iterable: list(filter(func, iterable)))
class Solution: def smallestBeautifulString(self, s: str, k: int) -> str: for i in range(len(s) - 1, -1, -1): no_exist = set()
for j in range(1, 3): if i - j >= 0: no_exist.add(s[i - j]) for j in range(1, 4): if std.letter_to_num(s[i]) + j + 1 <= k and chr( ord(s[i]) + j) not in no_exist: return self.gen_str(s, i, j) return ""
def gen_str(self, s: str, idx: str, offset: str) -> str: res = list(s) res[idx] = chr(ord(res[idx]) + offset)
for i in range(idx + 1, len(s)): no_exist = set() for j in range(1, 3): if i - j >= 0: no_exist.add(res[i - j]) for j in range(3): if chr(ord('a') + j) not in no_exist: res[i] = chr(ord('a') + j) break return ''.join(res)
|