| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 150530 | 이 대회 미분 되나요? (#200) | King of Chairs (FXCUP4_chairs) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <stdio.h>
#include <stdlib.h>
#include "king.h"
#include "vassal.h"
static int N, chk[101010];
static std::vector<int> W, C;
static void my_assert(int TF, const char* message){
if(!TF){ puts(message); exit(0); }
}
int main(){
my_assert(scanf("%d", &N) == 1, "Error: Invalid Input");
my_assert(1 <= N && N <= 100000, "Error: Invalid Input");
W.resize(N);
for(int i = 0; i < N; i++){
my_assert(scanf("%d", &W[i]) == 1, "Error: Invalid Input");
my_assert(1 <= W[i] && W[i] <= 1000000, "Error: Invalid Input");
}
C.resize(N);
for(int i = 0; i < N; i++){
my_assert(scanf("%d", &C[i]) == 1, "Error: Invalid Input");
my_assert(1 <= C[i] && C[i] <= 1000000, "Error: Invalid Input");
if(i) my_assert(C[i-1] <= C[i], "Error: Invalid Input");
}
long long num = SendInfo(W, C);
my_assert(0 <= num && num < (1ll << 60), "Wrong[1] : Invalid Number");
int chair_cnt = 0;
Init(num, C);
for(int i = 0; i < N; i++){
int ci = Maid(W[i]);
my_assert(-1 <= ci && ci < N, "Wrong[2] : Invalid Chair Number");
my_assert(-1 == ci || !chk[ci], "Wrong[2] : Invalid Chair Number");
my_assert(-1 == ci || W[i] <= C[ci], "Wrong[2] : Invalid Chair");
chk[ci] = 1;
if(ci>=0) chair_cnt++;
}
printf("Correct\n%d\n", chair_cnt);
return 0;
}
#include "vassal.h"
#include<vector>
#include<algorithm>
using namespace std;
#define SZ 1'000'000
long long BB;
int NN;
int seg[2100000];
vector<int> x[1000101];
int Update(int idx, int val, int n, int l, int r)
{
if (r < idx || idx < l) return seg[n];
if (l == r) return seg[n] += val;
int mid = l + r >> 1;
return seg[n] = Update(idx, val, n * 2, l, mid) + Update(idx, val, n * 2 + 1, mid + 1, r);
}
int Query(int L, int R, int n, int l, int r)
{
if (r < L || R < l) return 0;
if (L <= l && r <= R) return seg[n];
int mid = l + r >> 1;
return Query(L, R, n * 2, l, mid) + Query(L, R, n * 2 + 1, mid + 1, r);
}
int Kth(int k, int n, int l, int r)
{
if (l == r) return l;
int mid = l + r >> 1;
if (k <= seg[n * 2]) return Kth(k, n * 2, l, mid);
return Kth(k - seg[n * 2], n * 2 + 1, mid + 1, r);
}
void Init(long long B, std::vector<int> C){
NN = C.size();
BB = B;
for (int i = 0; i < NN; i++)
{
Update(C[i], 1, 1, 1, SZ);
x[C[i]].push_back(i);
}
sort(C.begin(), C.end());
if (B != NN) BB = C[B];
else BB = 1e9;
}
int Maid(int W){
if (W > BB) return -1;
if (Query(W, SZ, 1, 1, SZ) == 0) return -1;
int cnt = Query(0, W - 1, 1, 1, SZ) + 1;
int idx = Kth(cnt, 1, 1, SZ);
int ret = x[idx].back();
Update(idx, -1, 1, 1, SZ);
x[idx].pop_back();
return ret;
}