substring with concatenation of all words

17 篇文章 0 订阅

题目

You are given a string, s, and a list of words, words, that are all of the same length. Find all starting indices of substring(s) in s that is a concatenation of each word in words exactly once and without any intervening characters.

Example 1:

Input:
  s = "barfoothefoobarman",
  words = ["foo","bar"]
Output: [0,9]
Explanation: Substrings starting at index 0 and 9 are "barfoor" and "foobar" respectively.
The output order does not matter, returning [9,0] is fine too.

Example 2:

Input:
  s = "wordgoodgoodgoodbestword",
  words = ["word","good","best","word"]
Output: []

我的代码

package main

import "fmt"

func findSubstring(s string, words []string) []int {
	if words == nil || len(words) == 0 {
		return nil
	}
	var result = make([]int, 0)
	wordsMap := sliceToMap(words)
	wordLen := len(words[0])
	slen := len(s)

	temMap := make(map[string]int)
	for i := 0; i <= slen-wordLen*len(words); i++ {
		copyMap(wordsMap, temMap)
		matchWord := len(words)
		for j := i; j <= slen-wordLen; j += wordLen {
			word := s[j : j+wordLen]
			if value, ok := temMap[word]; ok && value > 0 {
				temMap[word] = value - 1
				matchWord--
				if matchWord == 0 {
					result = append(result, i)
					break
				}
			} else {
				break
			}

		}
	}
	return result
}

func sliceToMap(words []string) map[string]int {
	var wordsMap = make(map[string]int)
	for _, word := range words {
		wordsMap[word] = wordsMap[word] + 1
	}
	return wordsMap
}

func copyMap(src, target map[string]int) {
	for key, value := range src {
		target[key] = value
	}
}

func main() {
	fmt.Println(findSubstring("barfoothefoobarman", []string{"foo", "bar"}))
	fmt.Println(findSubstring("wordgoodgoodgoodbestword", []string{"word", "good", "best", "good"}))
	fmt.Println(findSubstring("foobarfoobar", []string{"foo", "bar"}))
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值