package main

import (
	"time"

	"math/rand"
)

// Initialize participants and workshops
func initializeParticipantsAndWorkshops(numParticipants, numWorkshops int) ([]Participant, []Workshop) {
	rand.Seed(time.Now().UnixNano())

	participants := make([]Participant, numParticipants)
	for i := range participants {
		prefs := rand.Perm(numWorkshops)[:3] // Pick 3 unique workshops
		participants[i] = Participant{
			ID:           i + 1,
			Preferences:  []int{prefs[0] + 1, prefs[1] + 1, prefs[2] + 1}, // Convert to 1-based index
			AssignedTo:   -1,
			RejectedFrom: make(map[int]bool),
		}
	}

	workshops := make([]Workshop, numWorkshops)
	for i := range workshops {
		workshops[i] = Workshop{
			ID:           i + 1,
			Capacity:     rand.Intn(10) + 15, // Capacity between 15 and 25
			Participants: []int{},
		}
	}

	return participants, workshops
}

// Main function
func main() {
	numParticipants := 600
	numWorkshops := 30

	participants, workshops := initializeParticipantsAndWorkshops(numParticipants, numWorkshops)
	startTime := time.Now()
	StableMatch(participants, workshops)
	duration := time.Since(startTime)
	PrintAssignments(participants, workshops)
	PrintStatistics(participants, workshops, duration)
}