[编程题]一封奇怪的信

[编程题]一封奇怪的信

现在你需要用一台奇怪的打字机书写一封书信。信的每行只能容纳宽度为100的字符，也就是说如果写下某个字符会导致行宽超过100，那么就要另起一行书写

信的内容由a-z的26个小写字母构成，而每个字母的宽度均会事先约定。例如字符宽度约定为[1,2,3,4,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5]，那么就代表'a'到'd'四个字母的宽度分别是1,2,3,4，而'e'到'z'的宽度均为5

那么按照上述规则将给定内容S书写成一封信后，这封信共有几行？最后一行宽度是多少？

输入描述:

输入为两行：
第一行是存储了每个字符宽度的字符串，包含26个数字，以1个空格分隔，每个数字均小于等于10
第二行是存储了待输入字符的字符串S，字符串S的长度在1到1000之间

输出描述:

输出为信的行数以及最后一行所包含的字符个数，中间以1个空格分隔

输入例子1:

5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
helloworld

输出例子1:

1 50

例子说明1:

"5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5"规定每个字符宽度为5
"helloworld"是输入的字符串S
由于S共包含10个字符，也即共占用50个字符宽度，因此可以写在同一行

输入例子2:

5 5 5 5 5 5 10 10 10 10 10 10 10 10 10 10 10 10 5 5 5 5 5 5 5 5
hahahahahahahaha

输出例子2:

2 20

例子说明2:

"5 5 5 5 5 5 10 10 10 10 10 10 10 10 10 10 10 10 5 5 5 5 5 5 5 5"规定了每个字符宽度
"hahahahahahahaha"是输入的字符串S
由于h宽度为10，a宽度为5，因此'hahahahahahah'占用100字符宽度可以写在第一行，‘aha’写在第二行也即最后一行。因此字符宽度为20

代码

#include<iostream>
#include<vector>
#include<string>
using namespace std;
int main()
{
    vector<int> v(26);
    for (int i = 0; i < 26; ++i)
        cin>> v[i];
    string s;
    cin>> s;
    long long nRow = 1, nWidth = 0;
    for (int i = 0; i < s.length()-1; ++i) {
        nWidth += v[s[i]-'a'];
        if (nWidth+v[s[i+1]-'a'] > 100) {
            nRow++;
            nWidth = 0;
        }
    }
    nWidth = nWidth+v[s[s.length()-1]-'a'];
    if (nWidth > 100)
        cout<< nRow+1<< " "<< nWidth - 100;
    else
        cout<< nRow<< " "<< nWidth;
    return 0;
}