import random

# 시뮬레이션 설정
num_simulations = 100  # 각 시뮬레이션당 가챠 횟수
num_repeats = 5000  # 시뮬레이션 반복 횟수

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

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

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

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

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

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

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

    # 무작위 확률로 캐릭터 등급 결정
    random_value = random.random()
    
    if random_value < probabilities["legendary"]:  # 전설 캐릭터 획득
        legendary_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, legendary_counter
    
    first_meeting_counter = 180
    legendary_counter = 100
    
    pickup_count = 0
    legendary_count = 0

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

    return pickup_count, legendary_count

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

for _ in range(num_repeats):
    pickup_count, legendary_count = simulate()
    total_pickup += pickup_count
    total_legendary += legendary_count

average_pickup = total_pickup / num_repeats
average_legendary = total_legendary / num_repeats
average_total_legendary = ( average_pickup + average_legendary ) / num_simulations

# 결과 출력
print(f"총 {num_repeats}회 반복 시뮬레이션 결과:")
print(f"시뮬레이션 당 {num_simulations}회 뽑기진행")
print(f"픽업 캐릭터 '{pickup_character}' 평균 획득 횟수: {average_pickup:.2f}")
print(f"그 외 전설 캐릭터 평균 획득 횟수: {average_legendary:.2f}")
print()
print(f"전설뽑기 확률: {average_total_legendary * 100:.2f}%")