2016-12-03 9 views
1

在下面的代码中,我很满意功能,即代码生成我期望的输出。但是,比较至Col至Int的长度 - 我有兴趣了解您是否可以提供某些内容(即至Col)。非常感谢!Excel列到Int,反之亦然 - 寻求改进

-- given a spreadsheet column as a string 
-- returns integer giving the position of the column 
-- ex: 
-- toInt "A" = 1 
-- toInt "XFD" = 16384 
toInt :: String -> Int 
toInt = foldl fn 0 
    where 
    fn = \a c -> 26*a + ((ord c)-64) 

-- given a integer returns 
-- the column to be found at that position as a [Char] 
-- ex: 
-- toCol 1 = "A" 
-- toCol 16384 = "XFD" 
toCol :: Int -> [Char] 
toCol n = toCol' n [] 
    where 
    toCol' 0 a = a 
    toCol' n a = 
     let r = mod n 26 in 
     case (r == 0) of 
      True -> toCol' (div (n-1) 26) ('Z':a) 
      False -> toCol' (div n 26) (chr(r + 64) : a) 
+0

建议无关你的问题:(1)你应该使用'foldl''(从'Data.List模块')而不是'toInt'中的'foldl' - 参见[* foldr与foldl(或foldl')*]的影响(http://stackoverflow.com/q/384797/2751851)。 (2)你可能想考虑使用'toInt :: String - > Maybe Int',这样如果给出一个无效的字符串(例如'“@#!”'),你就不会被迫返回无稽之谈。 – duplode

回答

2

每当你递归建立一个有限的名单,认为从Data.Listunfoldr :: (b -> Maybe (a, b)) -> b -> [a](尽管因为它是从错误的方向展开,我们最终需要扭转列表太)。语法扩展MultiWayIf也有助于使事情更好。

{-# LANGUAGE MultiWayIf #-} 

toCol :: Int -> [Char] 
toCol = reverse . unfoldr (\n -> let r = n `mod` 26 in 
      if | n == 0 -> Nothing 
      | r == 0 -> Just ('Z'   , n-1 `div` 26) 
      | otherwise -> Just (chr (r + 64), n `div` 26)) 

请注意,这也使得toCol点免费。如果你宁愿没有启用的扩展,宁愿模式匹配,你也可以这样做:

toCol :: Int -> [Char] 
toCol = reverse . unfoldr (\n -> case (n, n `mod` 26) of 
            (0, _) -> Nothing 
            (n, 0) -> Just ('Z'   , n-1 `div` 26) 
            (n, r) -> Just (chr (r + 64), n `div` 26)) 
+0

我的自行车风格偏好将会倾向于'=='上的模式匹配,所以'case mod 26 of ... r - > ...'。 – jberryman