제출 #1187898

#제출 시각아이디문제언어결과실행 시간메모리
1187898yhx3Stone Arranging 2 (JOI23_ho_t1)C++20
35 / 100
143 ms2260 KiB
#include <iostream>
#include <bits/stdc++.h>
#include <iomanip>
#include <numeric>
 
using namespace std;
 
#define ll long long
 
bool comp(const pair<int,int> &a,pair<int,int> &vl)
{
    return a.second > vl.first;
}
 
bool scomp(const pair<int,int> &a,pair<int,int> &vl)
{
    return a.first >= vl.first;
}
 

void solve() {
    int n;
    cin >> n;
    vector<int> a(n);
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
    }
    
    vector<int> ans(n);
    if (a[0] == 1) {
        bool ok = false;
        for (int i = n - 1; i >= 0; i--)
        {
            if (ok) {
                ans[i] = 1;
                continue;
            }
            if (a[i] == 1){
                ans[i] = 1;
                ok = true;
            } else {
                ans[i] = 2;
            }
        }
    } else {
        bool ok = false;
        for (int i = n - 1; i >= 0; i--)
        {
            if (ok) {
                ans[i] = 2;
                continue;
            }
            if (a[i] == 2){
                ans[i] = 2;
                ok = true;
            } else {
                ans[i] = 1;
            }
        }
    }
    for (int i = 0; i < n; i++)
    {
        cout << ans[i] << endl;
    }
    
}

int main() {
    int t = 1;
    // cin>>t;

    while(t--) {
        solve();
    }
    return 0;

}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...