Chefstr2.py — различия между версиями

Материал из DISCOPAL
Перейти к: навигация, поиск
 
(не показаны 2 промежуточные версии этого же участника)
Строка 1: Строка 1:
 
<source lang="python">
 
<source lang="python">
import sys
 
 
 
def main():
 
def main():
 
     symbols = {}
 
     symbols = {}
Строка 10: Строка 8:
 
         symbols[alphabet[i]] = i
 
         symbols[alphabet[i]] = i
 
      
 
      
     input_str = sys.stdin.readline()[:-1]
+
     input_str = input()
 
     n = len(input_str)
 
     n = len(input_str)
  
Строка 18: Строка 16:
 
     k_pow = 1e6
 
     k_pow = 1e6
 
     ind = 1
 
     ind = 1
     lim = 3 * n / 4
+
     lim = 0.55 * n  
 
     while ind <= lim:
 
     while ind <= lim:
        cur_diff = 0
 
 
         ks = -(n // -ind)  
 
         ks = -(n // -ind)  
         for i in range(ind):
+
 
            frequency = {}
+
        frequency = [0] * lenalf*ind               
            max_frequency = 0
+
        max_frequency = [0]*ind
             j = i
+
 
             while j < n:
+
         for i, idx in enumerate(input_idx):
                cur_freq = frequency.get(input_idx[j], 0) + 1
+
             j = i % ind
                if cur_freq > max_frequency:
+
             i_ = j*lenalf + idx
                    max_frequency = cur_freq
+
            cur_req = frequency[i_] + 1
                frequency[input_idx[j]] = cur_freq
+
            if cur_req > max_frequency[j]:
                j += ind
+
                max_frequency[j] = cur_req
            cur_diff += ks - max_frequency
+
            frequency[i_] = cur_req               
 +
 
 +
        cur_diff = ks*ind - sum(max_frequency)
 
   
 
   
 
         if cur_diff < operations:
 
         if cur_diff < operations:
Строка 42: Строка 41:
 
         ind += 1
 
         ind += 1
 
     print(f"{operations} {k_pow}")
 
     print(f"{operations} {k_pow}")
 
 
   
 
   
 
main()
 
main()
 
</source>
 
</source>

Текущая версия на 23:55, 4 мая 2022

def main():
    symbols = {}
    alphabet = "abcdefghijklmnopqrstuvwxyz"
    lenalf = len(alphabet)
 
    for i in range(lenalf):
        symbols[alphabet[i]] = i
 
    input_str = input()
    n = len(input_str)
 
    input_idx = bytearray([symbols[ch] for ch in input_str])
 
    operations = 1e6
    k_pow = 1e6
    ind = 1
    lim = 0.55 * n 
    while ind <= lim:
        ks = -(n // -ind) 
 
        frequency = [0] * lenalf*ind                
        max_frequency = [0]*ind
 
        for i, idx in enumerate(input_idx):
            j = i % ind
            i_ = j*lenalf + idx
            cur_req = frequency[i_] + 1
            if cur_req > max_frequency[j]:
                max_frequency[j] = cur_req
            frequency[i_] = cur_req                
 
        cur_diff = ks*ind - sum(max_frequency)
 
        if cur_diff < operations:
            operations = cur_diff
            k_pow = ks
        elif operations == cur_diff:
            if ks < k_pow:
                k_pow = ks
        ind += 1
    print(f"{operations} {k_pow}")
 
main()