import random

# 시뮬레이션 설정
num_simulations = 100  # 각 시뮬레이션당 가챠 횟수
num_repeats = 3000  # 시뮬레이션 반복 횟수
required_pickup_count = 1  # 최소 픽업 캐릭터 개수

# 가챠 확률 설정
probabilities = {
    "legendary": 0.02,   # 전설 캐릭터 확률
    "hero": 0.20,        # 영웅 캐릭터 확률
    "rare": 0.40,        # 희귀 캐릭터 확률
    "common": 0.38       # 일반 캐릭터 확률
}

# 전설 캐릭터 리스트
legendary_characters = ["베이라", "구즈만", "사만다", "미구엘", "자비에르", 
                        "파칼", "마이타", "콜", "디탈리오", "라비에", 
                        "릴리윌", "글로리아", "타이던", "노노윌", "레오니드", 
                        "이치", "네르갈", "눈갈", "이난나", "가르시아", 
                        "매그너스", "에다", "모모", "알렉세이", "시모나", 
                        "코코아"]

# 픽업 캐릭터 설정
pickup_character = "코코아"

# 확정 획득 카운터 설정
first_meeting_counter = 180
overall_counter = 100

def pull():
    """가챠 한 번을 실행하는 함수"""
    global overall_counter, first_meeting_counter

    first_meeting_counter -= 1  # 픽업 캐릭터 확정 카운터 감소
    overall_counter -= 1
    
    # 첫 만남 확정 획득 카운터가 0일 경우
    if first_meeting_counter == 0:
        first_meeting_counter = 180  # 첫 만남 카운터 초기화
        overall_counter = 100  # 전체 카운터 초기화
        return "pickup"  # 픽업 캐릭터 확정 획득

    # 전체 확정 획득 카운터가 0일 경우
    if overall_counter == 0:
        character = random.choice(legendary_characters)
        if character == pickup_character:
            first_meeting_counter = 180  # 첫 만남 카운터 초기화
            return "pickup"
        overall_counter = 100  # 전체 카운터 초기화
        return "legendary"  # 전설 캐릭터 확정 획득

    # 무작위 확률로 캐릭터 등급 결정
    random_value = random.random()
    
    if random_value < probabilities["legendary"]:  # 전설 캐릭터 획득
        overall_counter = 100  # 전체 확정 획득 카운터 초기화
        
        # 전설 캐릭터 중 픽업 여부 결정
        if random.random() < 0.5:
            first_meeting_counter = 180  # 첫 만남 카운터 초기화
            return "pickup"
        else:
            return "legendary"

    
    elif random_value < probabilities["legendary"] + probabilities["hero"]:  
        # 영웅 캐릭터 획득
        return "hero"
        
    elif random_value < probabilities["legendary"] + probabilities["hero"] + probabilities["rare"]: 
         # 희귀 캐릭터 획득
        return "rare"
        
    else: 
         # 일반 캐릭터 획득
        return "common"
        
def simulate():
    """하나의 시뮬레이션 실행 후 결과를 반환"""
    global first_meeting_counter, overall_counter
    
    first_meeting_counter = 180
    overall_counter = 100
    
    pickup_count = 0

    for _ in range(num_simulations):
        result = pull()
        if result == "pickup":
            pickup_count += 1

    return pickup_count

# 반복 시뮬레이션 실행 및 최소 픽업 캐릭터 횟수 계산
count_at_least_n = 0

for _ in range(num_repeats):
    pickup_count = simulate()
    if pickup_count >= required_pickup_count:
        count_at_least_n += 1

probability_at_least_n = count_at_least_n / num_repeats

# 결과 출력
print(f"총 {num_repeats}회 반복 시뮬레이션 결과:")
print(f"시뮬레이션 당 {num_simulations}회 뽑기진행")
print()
print(f"{pickup_character}를 적어도 {required_pickup_count}개 얻을 확률: {probability_at_least_n * 100:.2f}%")