EP-A Invasão.py

"""Após uma seleção muito rigorosa, você foi escolhido como o mais capacitado para a realização de uma tarefa de essencial importância para a continuidade da espécie humana.
CONFIDENCIAL: A Terra será invadida por extraterrestres com a intenção de total extermínio de seus habitantes para futura ocupação do planeta. Descobrir a data dessa invasão é de vital importância para impedir tamanha destruição. Informantes alienígenas aliados conseguiram obter o códifo de ataque e as regras para sua decodificação. Sua tarefa será desenvolver um programa que leia o código secreto e o decodifique, determinanndo a data da invasão.

### Mensagem dos Informantes
O código é composto por três partes: a primeira parte são dois números naturais m e n, a segunda parte é formada por m números naturais e a terceira por n números naturais.
Para determinar a data do evento, é necessário escolher o método correto: Método A ou Método B. Para saber o método adequado, existe o seguinte procedimento: (I) soma-se todos os m números, obtendo sm; (II) soma-se todos os dígitos de sm, obtendo sd; (III) se sd é par, aplica-se o Método A, caso contrário, aplica-se o Método B.

### Método A
Instruções: (I) ordena-se crescentemente os n números; (II) multiplica-se cada um dos n números por sua posição na sequência, considerando que o 1o item tem posição 1; (III) somam-se os n números já multiplicados, obtendo sn.

### Método B
Instruções: (I) soma-se a cada um dos n números o respectivo valor da sequência (1, 2, 4, 7, 11, 16, ...); (II) para cada n número já acrescido, tomam-se os dois dígitos mais à direita para serem acumulados; (III) o acúmulo total gera sn.

### Descobrindo a data
Descoberto o valor de sn, para determinar a data da invasão calcula-se o resto da divisão inteira de sn por 31, que representa o dia da invasão, adote zero como dia 31. De modo análogo, o resto da divisão inteira de sn por 12 indica o mês da invasão, o resto zero indicando dezembro. O ano, infelizmente, não pode ser descoberto.

### Observações
1) Faça a validação da entrada: (I) 0 < m < 6; (II) 0 < n < 11; (III) para todo x pertencente aos m números, 0 < x < 500.
2) Não use funções/métodos pré-definidos da linguagem para ordenar ou somar os itens das listas.
3) Considere que fevereiro possui 28 dias, pois os extraterrestres não atacam em anos bissextos.
4) Caso o procedimento resulte em um dia que excede o máximo para o mês descoberto, deve-se supor que o código foi corrompido. Uma mensagem deve indicar essa situação!
"""


def metodo_a(lista):
    # Ordenando a lista crescentemente
    for i in range(len(lista)):
        for j in range(len(lista)):
            if lista[i] < lista[j]:
                aux = lista[i]
                lista[i] = lista[j]
                lista[j] = aux

    # Multiplicando o número por sua posição
    for i in range(len(lista)):
        lista[i] = lista[i] * (i + 1)

    # Soma total
    sn = 0
    for i in range(len(lista)):
        sn = sn + lista[i]

    # Retorno da função
    return sn


def metodo_b(lista):
    seq = [1]
    sn = 0
    # Criando a sequência
    for i in range(1, len(lista)):
        seq.append(seq[i - 1] + i)
    # Somando os elementos da lista à seq, tomando os dois digitos mais a direita e somando os digitos
    for i in range(len(lista)):
        lista[i] = (lista[i] + seq[i]) % 100
        sn = sn + (lista[i] // 10) + (lista[i] % 10)

    # Retorno da função
    return sn


def entrada_m():
    m = int(input("m: "))

    while m > 6 and m < 0:
        m = int(input("m deve estar no intervalo 0 < m < 6.\nm: "))

    return m


def entrada_n():
    n = int(input("n: "))

    while n < 0 and n > 11:
        n = int(input("n deve estar no intervalo 0 < n < 11.\nn: "))

    return n


def entrada_x(tam: int, lista: list, nome_lista: str):
    for i in range(tam):
        lista.append(int(input(f"{nome_lista}[{i}]: ")))
        while lista[i] < 0 and lista[i] > 500:
            lista[i] = input(
                int(f"O número deve pertencer ao intervalo 0 < x < 500\n{nome_lista}[{i}]: "
                    ))


def seleciona_metodo(lista: list):
    # Soma dos valores da lista
    sm = 0
    for i in range(len(lista)):
        sm = sm + lista[i]

    # Soma dos dígitos da soma
    sd = 0
    while sm != 0:
        sd = sd + (sm % 10)
        sm = sm // 10

    if sd % 2 == 0:
        return 'A'

    return 'B'


def descobre_data(dia: int, mes: int):
    # Dicionário para nomes dos meses
    meses = {
        1: "janeiro",
        2: "feveiro",
        3: "março",
        4: "abril",
        5: "maio",
        6: "junho",
        7: "julho",
        8: "agosto",
        9: "setembro",
        10: "outubro",
        11: "novembro",
        12: "dezembro"
    }
    # Conferindo a validade do mês obtido
    if mes > 12 or dia > 31:
        print(f"Código corrompido! (mês= {mes}, dia={dia})")
        return
    # Conferindo a validade do dia para os mêses com < 30 dias
    elif mes % 2 == 0:
        # Conferindo a validade para fevereiro
        if mes == 2:
            if dia > 28:
                print(f"Código corrompido (mês= {mes}, dia={dia})")
                return
        # Conferindo a validade do dia para os outros meses de 30 dias
        elif dia > 30:
            print(f"Código corrompido (mês= {mes}, dia={dia})")
            return
    # Conferindo a validade da data para os outros meses:
    print(f"Invasão: {dia} de {meses[mes]}")


def main():
    m = entrada_m()
    n = entrada_n()
    lista_m = []
    lista_n = []
    entrada_x(m, lista_m, 'm')
    entrada_x(n, lista_n, 'n')
    if seleciona_metodo(lista_m) == 'A':
        print('Método A')
        sn = metodo_a(lista_n)
    else:
        print("Método B")
        sn = metodo_b(lista_n)

    dia = sn % 31
    if dia == 0:
        dia = 31
    mes = sn % 12
    if mes == 0:
        mes = 12
    descobre_data(dia, mes)


if __name__ == '__main__':
    main()