Count of lines required to write the given String

Given a string str and an integer array width[] where:

width[0] = width of character ‘a’
width[1] = width of character ‘b’

width[25] = width of character ‘z’

The task is to find the number of lines it’ll take to write the string str on a paper and the width of the last line upto which it is occupied.
Note: Width of a line is 10 units.

Examples:

Input: str = “bbbcccdddaa”,
width[] = {4, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}
Output: (2, 8)
“bbbcccddd” will cover first line (9 * 1 = 9 units)
As ‘a’ has a width of 4 which cannot fit the remaining 1 unit in the first line.
It’ll have to be written in the second line.
So, next line will contain “aa” covering 4 * 2 = 8 units.
We need 1 full line and one line with width 8 units.

Input: str = “abcdefghijklmnopqrstuvwxyz”,
width[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}
Output: (3, 6)
All the characters have the same width of 1. To write all 26 characters,
We need 2 full lines and one line with width 6 units.



Approach: We will write each character in the string str one by one. As we write a character, we immediately update (lines, width) that keeps track of how many lines we have used till now and what is the length of the used space in the last line.
If the width[char] in str fits our current line, we will add it. Otherwise, we will start with a new line

Below is the implementation of the above approach:

Python3

filter_none

edit
close

play_arrow

link
brightness_4
code

# Python3 implementation of the approach
  
# Function to return the number of lines required
def numberOfLines(S, widths):
  
    # If string is empty
    if(S == ""):
        return 0, 0
  
    # Initialize lines and width
    lines, width = 1, 0
  
    # Iterate through S
    for c in S:
        w = widths[ord(c) - ord('a')]
        width += w
        if width > 10:
            lines += 1
            width = w
  
    # Return lines and width used
    return lines, width
  
  
# Driver Code
S = "bbbcccdddaa"
Widths = [4, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
  
# Function call to print required answer
print(numberOfLines(S, Widths))

chevron_right


Output:

(2, 8)


My Personal Notes arrow_drop_up

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.