6

我要实现的是一个将字符串增加一个字符的函数，例如：

'AAA' + 1 = 'AAB'
'AAZ' + 1 = 'ABA'
'ZZZ' + 1 = 'AAAA'

我已经为前两种情况实现了功能，但是对于第三种情况我想不出任何解决方案。

这是我的代码：

def new_sku(s):
    s = s[::-1]
    already_added = False
    new_sku = str()

    for i in s:
        if not already_added:
            if (i < 'Z'):
                already_added = True
            new_sku += chr((ord(i)+1)%65%26 + 65)
        else:
            new_sku += i

    return new_sku[::-1]

有什么建议么 ？

4

10 回答 10

9

如果您正在处理双射计算，那么您可能有（或应该有）函数来转换为/从双射表示；将其转换为整数，递增，然后再转换回来会容易得多：

def from_bijective(s, digits=string.ascii_uppercase):
    return sum(len(digits) ** i * (digits.index(c) + 1)
               for i, c in enumerate(reversed(s)))

def to_bijective(n, digits=string.ascii_uppercase):
    result = []
    while n > 0:
        n, mod = divmod(n - 1, len(digits))
        result += digits[mod]
    return ''.join(reversed(result))

def new_sku(s):
    return to_bijective(from_bijective(s) + 1)
于 2013-07-31T09:10:24.933 回答
3

怎么样 ？

def new_sku(s):
    s = s[::-1]
    already_added = False
    new_sku = str()

    for i in s:
        if not already_added:
            if (i < 'Z'):
                already_added = True
            new_sku += chr((ord(i)+1)%65%26 + 65)
        else:
            new_sku += i

    if not already_added: # carry still left?
        new_sku += 'A'

    return new_sku[::-1]

样品运行：-

$ python sku.py Z
AA
$ python sku.py ZZZ
AAAA
$ python sku.py AAA
AAB
$ python sku.py AAZ
ABA
于 2013-07-31T09:03:12.030 回答
3

您必须将“AAA”、“ZZZ”……视为您操纵的价值的代表。

首先，解析值：

val = sum(pow(26, i) * (ord(v) - ord('A') + 1) for i, v in enumerate(value[::-1]))

然后，为其添加价值：

val = val + 1

编辑

最终值由下式给出：

res = ""
while val > 0:
     val, n = divmod(val - 1, 26)
     res = chr(n+ord('A')) + res

零表示的缺乏要求传递给 divmod 的值在每一轮递减，我还没有找到一种方法来处理列表理解。

编辑

而不是ord()and chr()，可以使用string.ascii_uppercase.index()andstring.ascii_uppercase[]

于 2013-07-31T09:17:48.163 回答
2

您可以在此处使用一些递归：

def new_sku(s):
    s = s[::-1]
    new_s = ''
    return expand(s.upper(), new_s)[::-1]

import string
chars = string.ascii_uppercase

def expand(s, new_s, carry_forward=True):

    if not s:
        new_s += 'A' if carry_forward else ''
        return new_s

    new_s += chars[(ord(s[0]) - ord('A') + carry_forward) % 26]

    # Slice the first character, and expand rest of the string
    if s[0] == 'Z': 
        return expand(s[1:], new_s, carry_forward)
    else:
        return expand(s[1:], new_s, False)


print new_sku('AAB')
print new_sku('AAZ')
print new_sku('ZZZ')
print new_sku('aab')
print new_sku('aaz')
print new_sku('zzz')

输出：

AAC
ABA
AAAA
AAC
ABA
AAAA
于 2013-07-31T09:28:29.450 回答
1

我会像带有进位的base-26加法一样实现它。

所以从字符串的右边开始，加 1。如果它到达 Z，则换行到 A 并将下一个最左边的字符向上移动一个。如果最左边的字符到达 Z，则在字符串的左侧添加一个 A。

s = ["Z","Z","Z"]
done = 0

index = len(s) - 1
while done == 0:
    if s[index] < "Z":
        s[index] = chr(ord(s[index]) + 1)
        done = 1
    else:
        s[index] = "A"
        if index == 0:
            s = ["A"] + s
            done = 1
        else:
            index = index - 1

print s
于 2013-07-31T09:04:51.183 回答
1

只需检查字符串是否都是Zs，如果是，则将其替换为长度为 s 的字符串len(s) + 1，仅由As 组成：

if s == "Z" * len(s):
    return "A" * (len(s) + 1)
于 2013-07-31T09:05:08.933 回答
1 
alp='ABCDEFGHIJKLMNOPQRSTUVWXYZA'
def rec(s):
    if len(s)==0:return 'A'
    last_letter=s[-1]
    if last_letter=='Z':return rec(s[:-1])+'A'
    return s[:-1]+alp[(alp.find(last_letter)+1)]

结果

>>> rec('AAA')
'AAB'
>>> rec('AAZ')
'ABA'
>>> rec('ZZZ')
'AAAA'
>>> rec('AZA')
'AZB'
于 2013-07-31T09:12:21.190 回答
1

这个怎么样？作为一种处理字符串变长的简单方法，您可以在前面加上一个前导 '@' 并在它没有增加的情况下将其剥离：

>>> def new_sku(s):
    def increment(s):
        if s.endswith('Z'):
            return increment(s[:-1])+'A'
        else:
            return s[:-1]+chr(ord(s[-1])+1)
    t = increment('@'+s)
    return t.lstrip('@')

>>> new_sku('AAA')
'AAB'
>>> new_sku('AAZ')
'ABA'
>>> new_sku('ZZZ')
'AAAA'

如果递归让您担心，那么您可以按照您已经做过的方式将其展平，但仍然使用添加和剥离的“@”字符。

于 2013-07-31T10:45:31.657 回答
0

您可以使用for-else循环：

from string import ascii_uppercase as au

def solve(strs):
    lis = []
    for i, c in enumerate(strs[::-1], 1):
        ind = au.index(c) + 2
        lis.append(au[(ind%26)-1])
        if ind <= 26:
            break
    else:
        # This will execute only if the for-loop didn't break.
        lis.append('A')

    return strs[:-1*i] + "".join(lis[::-1])

print solve('AAA')
print solve('AAZ')
print solve('ZZZ')
print solve('AZZZ')
print solve('ZYZZ')
print solve('ZYYZZ')

输出：

AAB
ABA
AAAA
BAAA
ZZAA
ZYZAA
于 2013-07-31T09:17:59.463 回答
0

我们可以看到总共有 3 个条件，您可以迭代字符串并处理其中一个条件。您可以使用 string.ascii_uppercase 代替 chr 和 ord

import string
def add(s):
    s = list(s)[::-1]
    for index, char in enumerate(s):
        if char != "Z":
            s[index] = string.ascii_uppercase[string.ascii_uppercase.index(char) + 1]
            return s[::-1]
        elif char == "Z" and (index != len(s) - 1):
            s[index] = "A"
        elif char == "Z" and (index == len(s) - 1):
            s[index] = "A"
            return ["A"] + s[::-1]
于 2013-08-04T13:43:56.117 回答