【Text Justification】cpp

题目:

Given an array of words and a length L, format the text such that each line has exactly L characters and is fully (left and right) justified.

You should pack your words in a greedy approach; that is, pack as many words as you can in each line. Pad extra spaces ' ' when necessary so that each line has exactly L characters.

Extra spaces between words should be distributed as evenly as possible. If the number of spaces on a line do not divide evenly between words, the empty slots on the left will be assigned more spaces than the slots on the right.

For the last line of text, it should be left justified and no extra space is inserted between words.

For example,
words: ["This", "is", "an", "example", "of", "text", "justification."]
L: 16.

Return the formatted lines as:

[
   "This    is    an",
   "example  of text",
   "justification.  "
]

 

Note: Each word is guaranteed not to exceed L in length.

click to show corner cases.

Corner Cases:

 

  • A line other than the last line might contain only one word. What should you do in this case?
    In this case, that line should be left-justified.

代码:

class Solution {
public:
    vector<string> fullJustify(vector<string>& words, int maxWidth) {
            vector<string> ret;
            vector<string> tmp; // words in one line
            int width = 0; // total width in one line
            const char BLANK=' ';
            int i=0;
            while ( i<words.size() )
            {
                // judge if a word can be added in a line
                if ( width+words[i].size() <= maxWidth )
                {
                    if ( width+words[i].size()==maxWidth )
                    {
                        tmp.push_back(words[i]);
                        width = maxWidth;
                    }
                    else
                    {
                        tmp.push_back(words[i]+string(1,BLANK));
                        width += words[i].size()+1;
                    }
                }
                // or create a new line & handle the words[i]
                else
                {
                    // CREAT A NEW LINE
                    // number of blank need to add
                    int blank_num =  maxWidth - width;
                    if ( tmp.back()[tmp.back().size()-1]==BLANK )
                    {
                        tmp.back() = string(tmp.back().begin(),tmp.back().end()-1);
                        blank_num = blank_num + 1;
                    }
                    // number of position for blanks
                    int pos_num = tmp.size()-1;
                    if ( pos_num==0 ) 
                    { 
                        ret.push_back(tmp.back()+string(blank_num, BLANK));
                    }
                    else
                    {
                        // number of blanks remain to be evenly distributed
                        int blank_remain = blank_num % pos_num;
                        for ( int j=0; j<tmp.size()-1; ++j )
                        {
                            tmp[j] = tmp[j]+ string(blank_num/pos_num, BLANK);
                        }
                        for ( int j=0; j<blank_remain; ++j )
                        {
                            tmp[j] = tmp[j] + string(1, BLANK);
                        }
                        string str = "";
                        for ( int j=0; j<tmp.size(); ++j ) str += tmp[j];
                        ret.push_back(str);
                    }
                    // HANDLE THE words[i]
                    tmp.clear();
                    if ( words[i].size()==maxWidth )
                    {
                        tmp.push_back(words[i]);
                        width = maxWidth;
                    }
                    else
                    {
                        tmp.push_back(words[i]+string(1, BLANK));
                        width = words[i].size()+1;
                    }
                }
                ++i;
            }
            // address the remain line
            if ( tmp.size()!=0 )
            {
                int blank_num =  maxWidth - width;
                string last_line = "";
                for ( int i=0; i<tmp.size(); ++i ) last_line += tmp[i];
                ret.push_back(last_line+string(blank_num, BLANK));
            }
            return ret;
    }
};

tips:

1. 要对string的各种操作都很熟悉

2. 要理解对题意,重点是到底啥是evenly:意思是不能evenly,就从左边到右一个个分配。

3. 扫一扫各种test case,多扫几遍可以AC了。

这道题自己扫了4次,终于AC了;经验就是,如果不能做到第一次把所有关键细节都考虑完全了,至少把一些细节(诸如blank_num,pos_num,blank_remain)单独列拎出来,这样做的好处就是如果遇上各种需要考虑的corner cases可以单独处理这些关键细节,而不用影响其他的部分。

====================================================

第二次过这道题,算是写过的leetcode中最繁琐的之一了,欣慰的是一次AC了。

class Solution {
public:
    vector<string> fullJustify(vector<string>& words, int maxWidth) {
            vector<string> ret;
            vector<string> tmp;
            int currWidth = 0;
            for ( int i=0; i<words.size()-1; ++i )
            {
                // no room for curr word
                if ( currWidth+words[i].size()>maxWidth )
                {
                    // address words already in tmp
                    int blank = maxWidth - currWidth + 1;
                    tmp.back() = tmp.back().substr(0,tmp.back().size()-1);
                    if ( tmp.size()>1 ){
                        int even = blank/(tmp.size()-1);
                        for ( int k=0; k<tmp.size()-1; k++ ) tmp[k] = tmp[k] + string(even,' ');
                        int remain = blank%(tmp.size()-1);
                        for ( int k=0; k<remain; k++ ) tmp[k] = tmp[k] + " ";
                        string str = "";
                        for ( int k=0; k<tmp.size(); ++k ) str += tmp[k];
                        ret.push_back(str);
                    }
                    else{
                        tmp[0] = tmp[0]+string(blank,' ');
                        ret.push_back(tmp[0]);                    
                    }
                    tmp.clear();
                    // consider words[i]'s width
                    if ( words[i].size()<maxWidth ) {
                        tmp.push_back(words[i]+" ");
                        currWidth = words[i].size()+1;
                    }
                    else{
                        ret.push_back(words[i]);
                        currWidth = 0;
                    }
                }
                // room for current word
                else 
                {
                    // no room for extra blank
                    if ( currWidth+words[i].size()==maxWidth )
                    {
                        tmp.push_back(words[i]);
                        string str = "";
                        for ( int k=0; k<tmp.size(); ++k ) str += tmp[k];
                        ret.push_back(str);
                        tmp.clear();
                        currWidth = 0;
                    }
                    // room for extra blank
                    else
                    {
                        tmp.push_back(words[i]+" ");
                        currWidth += words[i].size()+1;
                    }
                }
            }
            // last line
            if ( currWidth+words[words.size()-1].size()>maxWidth )
            {
                    // address words already in tmp
                    int blank = maxWidth - currWidth + 1;
                    tmp.back() = tmp.back().substr(0,tmp.back().size()-1);
                    if ( tmp.size()>1 ){
                        int even = blank/(tmp.size()-1);
                        for ( int k=0; k<tmp.size()-1; k++ ) tmp[k] = tmp[k] + string(even,' ');
                        int remain = blank%(tmp.size()-1);
                        for ( int k=0; k<remain; k++ ) tmp[k] = tmp[k] + " ";
                        string str = "";
                        for ( int k=0; k<tmp.size(); ++k ) str += tmp[k];
                        ret.push_back(str);
                    }
                    else{
                        tmp[0] = tmp[0]+string(blank,' ');
                        ret.push_back(tmp[0]);                    
                    }
                    // address the last line
                    blank = maxWidth - words[words.size()-1].size();
                    ret.push_back(words[words.size()-1]+string(blank,' '));
            }
            else
            {
                    tmp.push_back(words[words.size()-1]+string(maxWidth-currWidth-words[words.size()-1].size(),' '));
                    string str = "";
                    for ( int k=0; k<tmp.size(); ++k ) str += tmp[k];
                    ret.push_back(str);
            }
            return ret;
    }
};

 

posted on 2015-06-09 22:43  承续缘  阅读(196)  评论(0编辑  收藏  举报

导航