You are given a string
s
containing lowercase English letters, and a matrix shift
, where shift[i] = [direction, amount]
:direction
can be0
(for left shift) or1
(for right shift).amount
is the amount by which strings
is to be shifted.- A left shift by 1 means remove the first character of
s
and append it to the end. - Similarly, a right shift by 1 means remove the last character of
s
and add it to the beginning.
Return the final string after all operations.
Example 1:
Input: s = "abc", shift = [[0,1],[1,2]] Output: "cab" Explanation: [0,1] means shift to left by 1. "abc" -> "bca" [1,2] means shift to right by 2. "bca" -> "cab"
class Solution:
def stringShift(self, s: str, shift: List[List[int]]) -> str:
l,ls,rs = len(s),0,0
for direction , amount in shift:
if direction == 0:
ls+=amount
else:
rs+=amount
ls = ls % l
rs = rs % l
if ls == rs:
return s
elif ls > rs:
ls = ls -rs
return s[ls:]+s[:ls]
else:
rs = rs -ls
return s[-rs:]+s[:-rs]
No comments:
Post a Comment