Модель заражения группы наркоманов

// Наркоман
type Junkie struct {
    ID       int
    Infected bool
}
 
// Эксперимент
type Trial struct {
    Junkies []Junkie
}
 
func NewDefaultTrial() *Trial {
 
    t := Trial{}
 
    t.Junkies = append(t.Junkies, Junkie{Infected: true, ID: 0})
 
    for i := range 9 {
        t.Junkies = append(t.Junkies, Junkie{Infected: false, ID: i + 1})
    }
 
    return &t
}
 
func (t *Trial) ShootUpJunkies() {
 
    rand.Shuffle(len(t.Junkies), func(i, j int) {
        t.Junkies[i], t.Junkies[j] = t.Junkies[j], t.Junkies[i]
    })
 
    isNeedleInfected := false
 
    for i, junkie := range t.Junkies {
 
        if junkie.Infected {
            isNeedleInfected = true
        }
 
        if isNeedleInfected {
            t.Junkies[i].Infected = t.Junkies[i].Infected || (rand.Intn(100) < 2)
        }
    }
}
 
func (t *Trial) IsAllInfected() bool {
 
    var result bool
 
    result = true
 
    for _, j := range t.Junkies {
        result = result && j.Infected
    }
 
    return result
}

package main
 
import (
    "fmt"
 
    "github.com/alhaos/junkiesInection/internal/model"
)
 
const count = 1_000_000
 
func main() {
 
    var sum int
 
    for _ = range count {
        t := model.NewDefaultTrial()
        i := 0
        for ; !t.IsAllInfected(); i++ {
            t.ShootUpJunkies()
        }
        sum += i
    }
 
    fmt.Printf("avg days: %d", sum/count)
}