package rule
|
|
import (
|
"math/rand"
|
"time"
|
)
|
|
// 滑块组件-初始化并控制概率赋值1的二维数组
|
func initializeArrayWithProbability(rows, cols int, probability float64) [][]int {
|
array := make([][]int, rows)
|
rand.Seed(time.Now().UnixNano()) // 使用当前时间作为随机数种子
|
|
for i := range array {
|
array[i] = make([]int, cols)
|
for j := range array[i] {
|
// 生成0到99之间的随机数，如果小于等于概率*100，则赋值为1
|
if rand.Intn(100) <= int(probability*100) {
|
array[i][j] = 1
|
}
|
}
|
}
|
return array
|
}
|
|
// 滑块组件-统计滑动窗口内值为1的元素数量
|
func countOnesInWindow(window [][]int) int {
|
count := 0
|
for _, row := range window {
|
for _, val := range row {
|
if val == 1 {
|
count++
|
break // 一旦找到一个元素为1，跳出内层循环
|
}
|
}
|
}
|
return count
|
}
|
|
// 滑块组件-统计滑动窗口内至少有一个元素为1的行数
|
func countRowsWithOneInWindow(window [][]int) int {
|
count := 0
|
for _, row := range window {
|
for _, val := range row {
|
if val == 1 {
|
count++
|
break // 一旦找到一个元素为1，跳出内层循环
|
}
|
}
|
}
|
return count
|
}
|
|
// 滑块组件-统计最高行数以及滑块坐标
|
func CountMaxRows(modelMatrix [][]int, windowRows, windowCols int) (int, int, int) {
|
maxRowCountWindowCols := 0
|
maxRowCount := 0
|
// 统计滑动窗口内值为1的元素数量和至少有一个元素为1的行数
|
for i := 0; i <= 1440-windowCols; i++ {
|
window := make([][]int, windowRows)
|
for j := range window {
|
window[j] = modelMatrix[j][i : i+windowCols]
|
}
|
//onesCount := countOnesInWindow(window)
|
rowsWithOneCount := countRowsWithOneInWindow(window)
|
if rowsWithOneCount == 0 {
|
continue
|
}
|
if rowsWithOneCount > maxRowCount {
|
maxRowCount = rowsWithOneCount
|
maxRowCountWindowCols = i
|
}
|
|
//fmt.Printf("从第 %d 列到第 %d 列，值为1的元素数量为：%d，至少有一个元素为1的行数为：%d\n", i, i+windowCols-1, onesCount, rowsWithOneCount)
|
}
|
//fmt.Println("documentNumber: ", info.DocumentNumber)
|
//fmt.Println("maxRowCount: ", maxRowCount)
|
//fmt.Println("maxRowCountWindowCols: ", maxRowCountWindowCols, maxRowCountWindowCols+windowCols-1)
|
|
return maxRowCount, maxRowCountWindowCols, maxRowCountWindowCols + windowCols - 1
|
}
|
