【LeetCode】 806. 写字符串需要的行数

1.题目

806

我们要把给定的字符串 S 从左到右写到每一行上，每一行的最大宽度为100个单位，如果我们在写某个字母的时候会使这行超过了100 个单位，那么我们应该把这个字母写到下一行。我们给定了一个数组 widths ，这个数组 widths[0] 代表 ‘a’ 需要的单位， widths[1] 代表 ‘b’ 需要的单位，…， widths[25] 代表 ‘z’ 需要的单位。
现在回答两个问题：至少多少行能放下S，以及最后一行使用的宽度是多少个单位？将你的答案作为长度为2的整数列表返回。

2.思路

step1：定义变量用于存放行line和列wid
step2：遍历字符串中的每一个字母，把他对应的宽度相加
step3：如果大于100，则line++，wid=0，并且这个字母要成为下一行的首字母i–

3.代码

int line=0,wid=0;
  for(int i=0;i<S.length();i++){
  	if(wid+widths[S[i]-97]>100){
  		line++;
  		wid=0;
        i--;
	  }
	  else{
	  	wid+=widths[S[i]-97];
	  }
  }
  return {line+1,wid};

4.优秀案例

vector<int> res(2,0);
        int curLen=0,lines=1;
        for(auto i:S){
            curLen+=widths[i-'a'];
            if(curLen>100){
                ++lines;
                curLen=widths[i-'a'];
            }
        }
        res[0]=lines;
        res[1]=curLen;
        return res;
    }