제출 #1247623

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1247623		12baater	참나무 (IOI23_beechtree)	C++20	큐에 대기중	0 ms	0 KiB

beechtree

#include "beechtree.h"
#include <vector>
using namespace std;

vector<int> beechtree(int N, int M, std::vector<int> P, std::vector<int> C)
{
    bool all_same = 1;
    bool color = C[N-1];
    vector<int> ans(N);
    for (int i = N-1; i >= 0; i++) {
        if (all_same == 0) {
            ans[i] = 0;
        } else {
            if (C[i] == color) {
                ans[i] = 1;
            } else {
                all_same = 0;
                ans[i] = 0;
            }
        }
    }

    return ans;
}