import random

# 시뮬레이션 설정
num_repeats = 5000  # 시뮬레이션 반복 횟수
required_pickups = 1  # 목표 픽업 캐릭터 수

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

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

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

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

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_until_pickups(required_pickups):
    """목표 픽업 캐릭터 수까지 걸리는 뽑기 횟수를 반환"""
    global first_meeting_counter, overall_counter
    
    first_meeting_counter = 180
    overall_counter = 100
    
    pickup_count = 0
    non_pickup_legendary_count = 0
    pulls = 0

    while pickup_count < required_pickups:
        result = pull()
        pulls += 1
        if result == "pickup":
            pickup_count += 1
        elif result == "legendary":
            non_pickup_legendary_count += 1

    return pulls, non_pickup_legendary_count

# 반복 시뮬레이션 실행 및 평균 계산
total_pulls = 0
total_non_pickup_legendaries = 0

for _ in range(num_repeats):
    pulls, non_pickup_legendaries = simulate_until_pickups(required_pickups)
    total_pulls += pulls
    total_non_pickup_legendaries += non_pickup_legendaries

average_pulls = total_pulls / num_repeats
average_non_pickup_legendaries = total_non_pickup_legendaries / num_repeats

# 결과 출력
print(f"픽업 캐릭터 '{pickup_character}'가 적어도 {required_pickups}개 나올 때까지 평균 뽑기 횟수: {average_pulls:.2f}")
print(f"픽업 캐릭터 제외 전설 캐릭터 획득 평균 횟수: {average_non_pickup_legendaries:.2f}")