Austin/gkd
1
0
Fork 0
mirror of https://github.com/gkd-kit/gkd.git synced 2024-11-16 03:32:38 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

perf(selector): 优化 a<0 遍历逻辑

Browse Source
This commit is contained in:
lisonge 2023-11-15 20:31:01 +08:00
parent de2d99b1a6
commit 56949c2aa4
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
selector/src/commonMain/kotlin/li/songe/selector/data/PolynomialExpression.kt
View File

@ -1,7 +1,6 @@
package li.songe.selector.data
import li.songe.selector.NodeSequenceFc
import li.songe.selector.util.filterIndexes
/**
* an+b
@ -65,15 +64,15 @@ data class PolynomialExpression(val a: Int = 0, val b: Int = 1) : ConnectExpress
override val isConstant = numbers.size == 1
override val minOffset = (numbers.firstOrNull() ?: 1) - 1
private val b1 = b - 1
private val indexes = numbers.map { x -> x - 1 }
private val maxAb = a + b // when a<=0
override val traversal = object : NodeSequenceFc {
override fun <T> invoke(sq: Sequence<T?>): Sequence<T?> {
return if (a > 0) {
sq.filterIndexed { x, _ -> (x - b1) % a == 0 && (x - b1) / a > 0 }
return (if (a > 0) {
sq
} else {
sq.filterIndexes(indexes)
}
sq.take(maxAb)
}).filterIndexed { x, _ -> (x - b1) % a == 0 && (x - b1) / a > 0 }
}
}