68. Text Justification

https://leetcode.com/problems/text-justification/

JavaScript

/**
 * @param {string[]} words
 * @param {number} maxWidth
 * @return {string[]}
 */
var fullJustify = function(words, maxWidth) {
  var result = []
  var current = []
  var curLen = 0

  for (let i = 0; i < words.length; i++) {
    let word = words[i]

    if (curLen + word.length <= maxWidth) {
      current.push(word)
      curLen += word.length + 1
    } else {
      result.push(adjust(current, maxWidth))
      current = [word]
      curLen = word.length + 1
    }
  }

  var last = current.join(' ')
  var pads = maxWidth - last.length
  for (let i = 0; i < pads; i++) {
    last += ' '
  }
  result.push(last)

  return result

  function adjust(words, maxWidth) {
    var len = words.reduce((ret, word) => ret + word.length, 0)
    var pads = maxWidth - len

    if (words.length === 1) {
      for (let i = 0; i < pads; i++) {
        words.push(' ')
      }
      return words.join('')
    }

    var avg = pads / (words.length - 1) >> 0
    var mod = pads % (words.length - 1)
    var result = []

    for (let i = 0; i < words.length - 1; i++) {
      result.push(words[i])
      if (i < mod) {
        result.push(' ')
      }
      for (let j = 0; j < avg; j++) {
        result.push(' ')
      }
    }
    result.push(words[words.length - 1])

    return result.join('')
  }
}

Python

from functools import reduce


class Solution(object):

    def fullJustify(self, words, maxWidth):
        """
        :type words: List[str]
        :type maxWidth: int
        :rtype: List[str]
        """
        result = []
        current = []
        curlen = 0

        for word in words:
            if curlen + len(word) <= maxWidth:
                current.append(word)
                curlen += len(word) + 1
            else:
                result.append(self.adjust(current, maxWidth))
                current = [word]
                curlen = len(word) + 1

        last = ' '.join(current)
        pads = maxWidth - len(last)
        for i in range(pads):
            last += ' '
        result.append(last)

        return result

    def adjust(self, words, maxWidth):
        width = reduce(lambda a, b: a + len(b), words, 0)
        pads = maxWidth - width

        if len(words) == 1:
            for i in range(pads):
                words.append(' ')
            return ''.join(words)

        avg, mod = divmod(pads, len(words) - 1)
        result = []

        for i in range(len(words)-1):
            result.append(words[i])
            if i < mod:
                result.append(' ')
            for j in range(avg):
                result.append(' ')
        result.append(words[-1])

        return ''.join(result)