PTA刷题记录：L1-032 Left-pad (20分)

L1-032 Left-pad (20分)

题目描述： 根据新浪微博上的消息，有一位开发者不满NPM（Node Package Manager）的做法，收回了自己的开源代码，其中包括一个叫left-pad的模块，就是这个模块把javascript里面的React/Babel干瘫痪了。这是个什么样的模块？就是在字符串前填充一些东西到一定的长度。例如用去填充字符串GPLT，使之长度为10，调用left-pad的结果就应该是*****GPLT。Node社区曾经对left-pad紧急发布了一个替代，被严重吐槽。下面就请你来实现一下这个模块。

输入格式： 输入在第一行给出一个正整数N（≤10**4）和一个字符，分别是填充结果字符串的长度和用于填充的字符，中间以1个空格分开。第二行给出原始的非空字符串，以回车结束。

输出格式： 在一行中输出结果字符串。

输入样例1： 15 _ I love GPLT 输出样例1： ____I love GPLT 输入样例2： 4 * this is a sample for cut 输出样例2： cut

解题思路： 分情况对原字符串进行操作。若原字符串大于需要的字符数 n，则将原字符串截取为末尾 n 个字符；若原字符串小于需要的字符数 n，则向字符串头部加入缺少的数量个指定字符。 向字符串头部加入字符时，先将字符串翻转，此时将字符加入字符串尾部，加入完成后把字符串翻转回去，即完成了对字符串的头部加入。

通关代码：

#include <iostream>
#include <algorithm>

using namespace std;

void left_pad(int n, char ch, string &str) {
	if (str.size() < n) {
		reverse(str.begin(), str.end());
		
		int count = n - str.size();
		while (count--) {
			str += ch;
		}
		
		reverse(str.begin(), str.end());
		
	} else if (str.size() > n) {
		int pos = str.size() - n;
		
		str = str.substr(pos);
	}
}

int main() {
	int n;
	char ch;
	string str;
	
	cin >> n;
	cin >> ch;
	
	cin.ignore();
	getline(cin, str);
	
	left_pad(n, ch, str);
	
	cout << str;

	return 0;
}

通关截图：

在这里插入图片描述