D - Shortest and Longest LIS（构造+贪心）

Posted 2021-06-19 zjj0624

题意
给你一个字符串，由‘<’’>'组成，让你根据这个字符串构造两个由1~n组成的序列，如果s[i]=‘>’,就说明$a_i>a_{i+1}$，根据这个特性构造两个序列，一个序列的LIS是最小的，一个序列的LIS是最大的。
思路
刚拿到这个题，我就想到了贪心着来构造这两个序列，如果求LIS最大，那我们就尽量从小到大的选择数字，而且需要满足字符串的要求，另一个相反，但是想了好久怎么写代码，而且写的过程中调了很久。

#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 110;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n;
        cin>>n;
        vector<int>a,b;
        string s;
        cin>>s;
        int ans=n;
        set<int,greater<int>>ss;
        for(int i=1 ; i<=n ; i++) ss.insert(i);
        for(int i=0 ; i<s.length() ; i++) //求 LIS最小 从大往小放
        {
            if(s[i]=='>')
            {
                a.push_back(*ss.begin());
                ss.erase(ss.begin());
            }
            else
            {
                int cnt=0;
                for(int j=i ; j<s.length() ; j++)
                {
                    if(s[j]=='<') cnt++;
                    else break;
                }
                int c=*ss.begin();
                for(int j=cnt ; j>=1 ; j--)
                {
                    i++;
                    a.push_back(c-j);
                    ss.erase(c-j);
                }
                i--;
            }
        }
        while(!ss.empty())
        {
            a.push_back(*ss.begin());
            ss.erase(ss.begin());
        }
        set<int>se;
        for(int i=1 ; i<=n ; i++) se.insert(i);
        for(int i=0 ; i<s.length() ; i++)
        {
            if(s[i]=='<')
            {
                b.push_back(*se.begin());
                se.erase(se.begin());
            }
            else
            {
                int cnt=0;
                for(int j=i ; j<s.length() ; j++)
                {
                    if(s[j]=='>') cnt++;
                    else break;
                }
                int c=*se.begin();
                for(int j=cnt ; j>=1 ; j--)
                {
                    i++;
                    b.push_back(c+j);
                    se.erase(c+j);
                }
                i--;
            }
        }
        while(!se.empty())
        {
            b.push_back(*se.begin());
            se.erase(se.begin());
        }
        for(auto x:a) cout<<x<<" ";
        cout<<endl;
        for(auto x:b) cout<<x<<" ";
        cout<<endl;
    }
    return 0;
}