时间戳

首页 > 解决方案 > 在 Kotlin-way 中获取字符串中包含的子字符串的索引

kotlin - 在 Kotlin-way 中获取字符串中包含的子字符串的索引

问题描述

我想实现一个函数，它将返回指定字符串中子字符串的索引。现在我是用 Java 风格做的：

public fun String?.indexesOf(substr: String, ignoreCase: Boolean = true): List<Int> {
    var list = mutableListOf<Int>()
    if (substr.isNullOrBlank()) return list
    var count = 0;
    this?.split(substr, ignoreCase = ignoreCase)?.forEach {
        count += it.length
        list.add(count)
        count += substr.length
    }
    list.remove(list.get(list.size-1))
    return list
}

但我不认为这是一个 kotlin 方式的解决方案。它看起来最像典型的 java 程序，但用 kotlin 编写。如何使用 kotlin 更优雅地实现这一点？

标签： kotlin

解决方案

我会做什么如下：

fun ignoreCaseOpt(ignoreCase: Boolean) = 
    if (ignoreCase) setOf(RegexOption.IGNORE_CASE) else emptySet()

fun String?.indexesOf(pat: String, ignoreCase: Boolean = true): List<Int> =
    pat.toRegex(ignoreCaseOpt(ignoreCase))
        .findAll(this?: "")
        .map { it.range.first }
        .toList()

// check:
println("xabcaBd".indexesOf("ab", true))
println("xabcaBd".indexesOf("ab", false))
println("xabcaBd".indexesOf("abx", true))

val s: String? = null
println(s.indexesOf("aaa"))

// output:
[1, 4]
[1]
[]
[]

推荐阅读