#!/bin/env python
# -*- coding: utf-8 -*-

"""

给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。

注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。


示例 1：

输入：
  s = "barfoothefoobarman",
  words = ["foo","bar"]
输出：[0,9]
解释：
从索引 0 和 9 开始的子串分别是 "barfoor" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。
示例 2：

输入：
  s = "wordgoodgoodgoodbestword",
  words = ["word","good","best","word"]
输出：[]

"""


def find(s, words):
    res = []

    meta_map = {}
    for i in words:
        meta_map[i] = meta_map.get(i, 0) + 1

    length = len(words[0])
    fullscore = len(words)

    for i in xrange(len(s)):
        temp_map = {}
        score = 0

        for j in xrange(fullscore):
            x = i + j * length
            sub = s[x:x + length]
            if sub not in meta_map:
                break

            temp_map[sub] = temp_map.get(sub, 0) + 1
            if temp_map[sub] <= meta_map[sub]:
                score += 1
            else:
                break

        if score == fullscore:
            res.append(i)

    return res


print find("barfoothefoobarman", ["foo", "bar"])
print find("wordgoodgoodgoodbestword", ["word", "good", "best", "word"])
