def jumlah_korek_api(n):
    # Menggunakan rumus T(n) = 3 * n * (n + 1) // 2
        return 3 * n * (n + 1) // 2

        # Input tinggi segitiga
        tinggi_segitiga = int(input("Masukkan tinggi segitiga (minimal 1, maksimal 1000): "))

        if 1 <= tinggi_segitiga <= 1000:
            total_korek_api = jumlah_korek_api(tinggi_segitiga)
                print(f"Jumlah batang korek api yang diperlukan untuk tinggi {tinggi_segitiga} adalah {total_korek_api}") 
                else:
                    print("Tinggi harus antara 1 dan 1000!")