diff options
Diffstat (limited to 'challenge-064/paulo-custodio/python/ch-2.py')
| -rw-r--r-- | challenge-064/paulo-custodio/python/ch-2.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/challenge-064/paulo-custodio/python/ch-2.py b/challenge-064/paulo-custodio/python/ch-2.py new file mode 100644 index 0000000000..12034bea57 --- /dev/null +++ b/challenge-064/paulo-custodio/python/ch-2.py @@ -0,0 +1,52 @@ +#!/usr/bin/env perl + +# Challenge 064 +# +# TASK #2 > Word Break +# Submitted by: Mohammad S Anwar +# +# You are given a string $S and an array of words @W. +# +# Write a script to find out if $S can be split into sequence of one or more +# words as in the given @W. +# +# Print the all the words if found otherwise print 0. +# +# Example 1: +# Input: +# +# $S = "perlweeklychallenge" +# @W = ("weekly", "challenge", "perl") +# +# Output: +# +# "perl", "weekly", "challenge" +# Example 2: +# Input: +# +# $S = "perlandraku" +# @W = ("python", "ruby", "haskell") +# +# Output: +# +# 0 as none matching word found. + +import itertools +import unittest + +def word_break(S, *W): + k = len(W) + for words in itertools.permutations(W): + if ''.join(words) == S: + return ' '.join(W) + return "0" + +class TestWordBreak(unittest.TestCase): + def test_word_break(self): + self.assertEqual(word_break("perlweeklychallenge", "weekly", "challenge", "perl"), + "weekly challenge perl") + self.assertEqual(word_break("perlandraku", "python", "ruby", "haskell"), + "0") + +if __name__ == '__main__': + unittest.main() |
