Lukuvälit

#
#
# todo: multiprocessing
from multiprocessing import Pool
from math import ceil
from os import environ, popen
from sys import platform
 
 
def ydinten_määrä():
    """Palauttaa ydinten määrän."""
    if platform == 'win32':
        return int(environ['NUMBER_OF_PROCESSORS'])
    else:
        return int(popen('grep -c cores /proc/cpuinfo').read())
 
 
def kysy():
    """
    Kysytään luvut
    :return:
    """
    with Pool() as p:
        binäärit = p.map(binääriksi, range(19))
    syote_maara = int(input())
 
    valit = []
    for _ in range(syote_maara):
        syote = input().split()
        valit.append(
            ((int(syote[0]), int(syote[1])), binäärit)
        )
    return valit
 
 
# def tarkista(vali):
#     # Tarkistetaan kaikki luvut väliltä
#     count = 0
#     for n in range(vali[0], vali[1] + 1):
#         luku = str(n)
#
#         for numero in luku:
#             if numero not in ("0", "1"):
#                 sisältyy = False
#                 break
#             else:
#                 sisältyy = True
#         if sisältyy:
#             count += 1
#     return count
def binäärihaku(lista, haettava):
    """
    Etsii listasta alkion, jonka binäärin kymmenkantaiseksi tulkittu
    kokonaisluku on sama kuin haettava tai juuri haettavaa alempi.
 
    :param lista: Lista kymmenkantaisiksi tulkittuja binäärejä
    :param haettava: Kokonaislukuna haettava kohde
    :return:
    """
    # Suoritetaan binäärihaku välillä [a,b], missä a ja b ovat listan
    # alkioita
    a = 0
    b = len(lista) - 1
 
    while True:
        keski = round((a + b) / 2)
        # print("a", a)
        # print("b", b)
        # print("k", keski)
        # print()
 
        # Onko haettavan ympärillä vain kaksi indeksiä
        if b - a <= 1:
            return a
 
        # Haettava on välillä [a,keski[
        if haettava < lista[keski]:
            b = keski
            # print("<-")
        # Haettava on välillä ]keski,b]
        elif lista[keski] < haettava:
            a = keski
            # print("->")
        # Haettava on keski
        else:
            return keski
 
 
def määrät_alemmista(numeroinen):
    """
    Laskee alemmat binäärinkaltaiset luvut.
    :param numeroinen:
    :return:
    """
    # Lasketaan 0 mukaan
    count = 1
    for i in range(numeroinen):
        count += 2**i
    return count
 
 
def binääriksi(numeroinen):
    """
    Kaikki binäärit n-numeroisessa luvussa.
    Palauttaa listan n-numeroisista binääreistä, jotka on tulkittu
    kokonaisluvuiksi.
 
    :param numeroinen:
    :return: list
    """
    # Pienin n-numeroinen binääri
    binääri = int("1" + "0" * (numeroinen - 1), 2)
    binäärit = []
 
    # Binäärit listaan
    while True:
        # Tulkitaan binääri kokonaislukuna
        kokonais = int(f"{binääri:b}")
        binäärit.append(kokonais)
 
        # Toinen binääri
        binääri += 1
 
        if len(f"{binääri:b}") > numeroinen:
            break
    return binäärit
 
 
def tarkista(luku, binäärijoukot):
    """
    Lasketaan binääriä käyttäen.
    :param luku:
    :param binäärijoukot:
    :return count: 1 ja 0 sisältävien lukujen määrä
    """
    count = 0
 
    # Numeroiden määrä
    num_lkm = len(str(luku))
 
    # Suurin määrä n-numeroisessa luvussa
    määrien_ylä = int("1" * num_lkm)
    if luku >= määrien_ylä:
        count = määrät_alemmista(num_lkm)
    # Epäpositiiviset luvut rikkovat koko jutun
    elif luku <= 0:
        if luku == 0:
            count = 1
        else:
            count = 0
    else:
        # Tutkitaan tapauskohtaisesti
        #############################
 
        # Alempien numeroiden määrä
        count += määrät_alemmista(num_lkm - 1)
 
        # binäärilista = binääriksi(num_lkm)
        # print("Binäärilista:")
        # for i in range(len(binäärilista)):
        #     print(f"{binäärilista[i]:b}")
 
        # Suoritetaan binäärihaku, joka antaa binäärilistan alkion; lisätään 1
        count += binäärihaku(binäärijoukot[num_lkm], luku) + 1
 
    # print("Num_lkm:", num_lkm)
    # print("Binääri:", määrien_ylä)
    # print("Count:", count)
    # print()
    return count
 
 
# if __name__ == '__main__':
#     with Pool() as p:
#         BINÄÄRIT = p.map(binääriksi, range(19))
 
 
def määräfunktio(inp):
    väli, binäärit = inp
    return tarkista(väli[1], binäärit) - tarkista(väli[0] - 1, binäärit)
 
 
def main():
    # print(välit[0])
    ytimet = ydinten_määrä()
    print(ytimet)
 
    with Pool() as p:
 
        välit = kysy()
        for vastaus in p.imap(määräfunktio, välit, chunksize=ceil(len(välit)/ytimet)):
            print(vastaus)
 
    # määräfunktio = määräfunktio_asettaja(binäärit)
    # print(määräfunktio([0,10]))
 
 
if __name__ == '__main__':
    main()
 

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 4

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 5

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 6

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 8

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 9

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 10

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 11

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 12

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 13

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 18

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 19

Group: 2, 3

Verdict: WRONG ANSWER

Test 20

Group: 2, 3

Verdict: WRONG ANSWER

Test 21

Group: 3

Verdict: RUNTIME ERROR

multiprocessing.pool.RemoteTraceback: 
"""
Traceback (most recent call last):
  File "/usr...

Test 22

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 25

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 26

Group: 3

Verdict: RUNTIME ERROR

multiprocessing.pool.RemoteTraceback: 
"""
Traceback (most recent call last):
  File "/usr...

Test 27

Group: 3

Verdict: RUNTIME ERROR

multiprocessing.pool.RemoteTraceback: 
"""
Traceback (most recent call last):
  File "/usr...

Test 28

Group: 3

Verdict: RUNTIME ERROR

multiprocessing.pool.RemoteTraceback: 
"""
Traceback (most recent call last):
  File "/usr...