字符串选择

最新推荐文章于 2023-02-12 20:00:00 发布

翻译最新推荐文章于 2023-02-12 20:00:00 发布 · 417 阅读

leetcode 专栏收录该内容

176 篇文章

订阅专栏

本文介绍了一个用于判断两个字符串是否通过旋转得到对方的算法。通过将一个字符串与其自身连接，然后检查另一个字符串是否为这个新字符串的子串来实现。使用Python unittest进行测试，验证了不同情况下的正确性。

import unittest


def is_substring(string, sub):
    return string.find(sub) != -1


def string_rotation(s1, s2):
    if len(s1) == len(s2) != 0:
        return is_substring(s1 + s1, s2)
    return False


class Test(unittest.TestCase):
    '''Test Cases'''
    data = [
        ('waterbottle', 'erbottlewat', True),
        ('foo', 'bar', False),
        ('foo', 'foofoo', False)
    ]

    def test_string_rotation(self):
        for [s1, s2, expected] in self.data:
            actual = string_rotation(s1, s2)
            self.assertEqual(actual, expected)

if __name__ == "__main__":
    unittest.main()