418. 屏幕可显示句子的数量

Posted 2020-11-27 fdwzy

题目：

链接：https://leetcode-cn.com/problems/sentence-screen-fitting/

给你一个 rows x cols 的屏幕和一个用 非空 的单词列表组成的句子，请你计算出给定句子可以在屏幕上完整显示的次数。

注意：

1. 一个单词不能拆分成两行。
2. 单词在句子中的顺序必须保持不变。
3. 在一行中 的两个连续单词必须用一个空格符分隔。
4. 句子中的单词总量不会超过 100。
5. 每个单词的长度大于 0 且不会超过 10。
6. 1 ≤ rows, cols ≤ 20,000.

 

示例 1：

输入：
rows = 2, cols = 8, 句子 sentence = ["hello", "world"]

输出：
1

解释：
hello---
world---

字符 ‘-‘ 表示屏幕上的一个空白位置。

 

示例 2：

输入：
rows = 3, cols = 6, 句子 sentence = ["a", "bcd", "e"]

输出：
2

解释：
a-bcd- 
e-a---
bcd-e-

字符 ‘-‘ 表示屏幕上的一个空白位置。

 

示例 3：输入：rows = 4, cols = 5, 句子 sentence = ["I", "had", "apple", "pie"]

输出：
1

解释：
I-had
apple
pie-I
had--

字符 ‘-‘ 表示屏幕上的一个空白位置。
解答：

这真的是中等题??？我开始想了个动态规划，dp[i]表示第i个单词作为行首，最终句子的末尾元素列号。然后写了60
几行写不出来。

正确做法是用贪心法，先顺序放入句子。记录每次放完一轮句子，最后一个单词的末尾列号m，假设此时是k1行。如果
之后的遍历中，也是当句子遍历完一轮的时候，最后一个单词的末尾序号也是m，假设此时是k2行（当然k2>k1）。
那么说明k1行m列开始到k2行m列结束，一共有整数个完整的句子。那么往下直接跳过(k2-k1)行到达第k2+(k2-k1)
行的第m列，正好还是另一个句子的结尾。就这样一直往后跳，直到跳到不能再跳（剩余行数不够跳了），再
正常顺序放入句子直到不能再放为止。

 

 1 class Solution {
 2 public:
 3     int wordsTyping(vector<string>& sentence, int rows, int cols) {
 4         unordered_map<int,pair<int,int>> mp;//mp[j]=pair：放入pair[0]个句子、pair[1]行的最后元素的列数为j
 5         int cur_row=0,cur_col=0,words=sentence.size(),res=0;
 6         while(cur_row<rows){
 7             // cout<<cur_row<<" "<<cur_col<<endl;
 8             int i=0;
 9             while(cur_row<rows and i<words){
10                 if(cur_col<cols and cols-cur_col>=sentence[i].size()){
11                     cur_col+=sentence[i++].size()+1;
12                 }
13                 else{//换行
14                     ++cur_row;
15                     cur_col=0;
16                 }
17             }
18             if(i<words){
19                 return res;
20             }
21             ++res;
22             if(mp.count(cur_col-1)==0){
23                 mp[cur_col-1]={res,cur_row};
24             }
25             else{
26                 int row_dif=cur_row-mp[cur_col-1].second;
27                 int sen_dif=res-mp[cur_col-1].first;
28                 int p=(rows-1-cur_row)/row_dif;
29                 cur_row+=p*row_dif;
30                 res+=p*sen_dif;
31                 mp[cur_col-1]={res,cur_row};
32             }
33         }
34         return res;
35     }
36 };