AcWing-4715. 构造数组

链接：4715. 构造数组 - AcWing题库

请你构造一个长度为 n 的正整数数组 a1,a2,…,an。

我们会给出一个长度为 n−1 的由 <、>、= 组成的字符串 s1s2…sn−1 用于约束你的构造：

如果 si 为 <，则表示你构造的数组需满足 ai<ai+1。
如果 si 为 >，则表示你构造的数组需满足 ai>ai+1。
如果 si 为 =，则表示你构造的数组需满足 ai=ai+1。
你构造的正整数数组需满足上述约束的同时，保证 a1+a2+…+an 的值尽可能小。

请你输出满足条件的正整数数组。数据保证一定有解。

输入格式

第一行包含整数 n。

第二行包含字符串 s1s2…sn−1。

输出格式

共一行，输出 a1,a2,…,an。

数据范围

前 3 个测试点满足 2≤n≤6。
所有测试点满足 2≤n≤1000。

输入样例1：

5
><><

输出样例1：

2 1 2 1 2

输入样例2：

5
=<<<

输出样例2：

1 1 2 3 4

贪心+模拟题。由于数组每两相邻元素之间仅包含等于，大于，小于三种关系，那么我们可以把数组分成若干个非递增和非递减的连续序列。

而非递增连续序列的左边的值必然大于等于右边的值，非递减连续序列右边的值必然大于等于左边的值，那么只要确定非递增连续序列最右边的值，和非递减连续序列最左边的值就可以逐步构造出整个数组了。 

而由于题目要求数组所有元素和的值应该尽可能小，所以只要让非递增连续序列的最小值(最右边的值)和非递减连续序列的最小值(最左边的值)是1即可。  

代码如下：

#include<iostream>
using namespace std;
const int N = 1005;
int n;
char s[N]; //存符号的字符串 
int res[N] {0}; //答案数组，从0开始存数 
int main()
{
	scanf("%d", &n);
	scanf("%s", s + 1); //字符串从1开始存，为了方便构造数组，res数组第i位置上的数对应第i位置字符右边的数
	
	//找非递减的连续序列 
	for(int i=1; i<n; )
		if(s[i]=='<')
		{ 
			res[i-1] = 1; //非递减序列的最左端的值初始化为1
			while(s[i]!='>' && i<n)
			{
				if(s[i]=='<') res[i] = res[i-1] + 1; 
				else res[i] = res[i-1];
				i ++;
			}
		}
		else i ++;
	
	//找非递增的连续序列，为了构造方便，从右往左找 
	for(int i=n-1; i>=1; ) 
		if(s[i]=='>' || s[i]=='=') //把'='加入判断条件是为了处理没有'>'，却还有'='的右边位置没数的情况，如样例2. 
		{ 
			if(s[i]=='>') res[i] = 1; //非递增序列的最右端的值初始化为1
			else res[i] = (res[i-1]==0? 1:res[i-1]); //如果是'='，那么得判断其左边位置有没有数，没则赋值为1，反之为那个数 
			while(s[i]!='<' && i>=1)
			{
				if(s[i]=='>' && res[i-1]<=res[i]) res[i-1] = res[i] + 1;
				else if(s[i]=='=') res[i-1] = res[i];
				i --;
			}
		}
		else i --;

	for(int i=0; i<n; i++) cout << res[i] << " ";
	cout << endl;
	return 0; 
}