제출 #796288

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
796288		khshg	앵무새 (IOI11_parrots)	C++14	컴파일 에러	0 ms	0 KiB

encoder

#include "encoder.h"
#include "encoderlib.h"
#include <bits/stdc++.h>
using namespace std;

const int BASE = 3333;
int LEN;

struct BIGNUM {
	vector<int> d;
}dp[5 * 64][256];

BIGNUM add(const BIGNUM& A, const BIGNUM& B) {
	BIGNUM res;
	int C = 0;
	for(int i = 0; i < max((int)A.d.size(), (int)B.d.size()) || C; ++i) {
		int cur = C;
		if(i < (int)A.d.size()) cur += A.d[i];
		if(i < (int)B.d.size()) cur += B.d[i];
		C = cur >= BASE;
		if(C) cur -= BASE;
		res.d.push_back(cur);
	}
	return res;
}

BIGNUM sub(BIGNUM A, const BIGNUM& B) {
	int C = 0;
	for(int i = 0; i < (int)B.d.size() || C; ++i) {
		A.d[i] -= C + (i < (int)B.d.size() ? B.d[i] : 0);
		C = A.d[i] < 0;
		if(C) A.d[i] += BASE;
	}
	while((int)A.d.size() && A.d.back() == 0) A.d.pop_back();
	return A;
}

BIGNUM mul(const int& A, const BIGNUM& B) {
	BIGNUM res;
	int C = 0;
	for(int i = 0; i < (int)B.d.size() || C; ++i) {
		int cur = C;
		if(i < (int)B.d.size()) cur += A * B.d[i];
		C = cur / BASE;
		res.d.push_back(cur - C * BASE);
	}
	return res;
}

bool cmp(const BIGNUM& A, const BIGNUM& B) {
	if((int)A.d.size() != (int)B.d.size()) return (int)A.d.size() < (int)B.d.size();
	for(int i = (int)A.d.size(); i >= 0; ++i) {
		if(A.d[i] != B.d[i]) return A.d[i] < B.d[i];
	}
	return 0;
}
 
void encode(int N, int M[]) {
	LEN = 5 * N;
	for(int i = 0; i < 256; ++i) dp[0][i].d = {1};// ONE
	for(int i = 1; i < LEN; ++i) dp[i][0].d = {1};// ONE
	for(int i = 1; i < LEN; ++i) {
		for(int j = 1; j < 256; ++j) {
			dp[i][j] = add(dp[i][j - 1], dp[i - 1][j]);
		}
	}
	BIGNUM val, pwi;
	pwi.d = {1};
	for(int i = 0; i < N; ++i) {
		val = add(val, mul(M[i], pwi));
		if(i + 1 < N) pwi = mul(256, pwi);
	}
	for(int i = LEN - 1; i >= 0; --i) {
		int L = 0, R = 255;
		while(L < R) {
			int M = (L + R) / 2;
			if(cmp(dp[i][M], val)) {
				L = M + 1;
			} else {
				R = M;
			}
		}
		send(L);
		if(L) val = sub(val, dp[i][L - 1]);
	}
}

decoder

#include "decoder.h"
#include "decoderlib.h"
#include <bits/stdc++.h>
using namespace std;

const int BASE = 3333;
int LEN;

struct BIGNUM {
	vector<int> d;
}dp[5 * 64][256];

BIGNUM add(const BIGNUM& A, const BIGNUM& B) {
	BIGNUM res;
	int C = 0;
	for(int i = 0; i < max((int)A.d.size(), (int)B.d.size()) || C; ++i) {
		int cur = C;
		if(i < (int)A.d.size()) cur += A.d[i];
		if(i < (int)B.d.size()) cur += B.d[i];
		C = cur >= BASE;
		if(C) cur -= BASE;
		res.d.push_back(cur);
	}
	return res;
}

BIGNUM sub(BIGNUM A, const BIGNUM& B) {
	int C = 0;
	for(int i = 0; i < (int)B.d.size() || C; ++i) {
		A.d[i] -= C + (i < (int)B.d.size() ? B.d[i] : 0);
		C = A.d[i] < 0;
		if(C) A.d[i] += BASE;
	}
	while((int)A.d.size() && A.d.back() == 0) A.d.pop_back();
	return A;
}

BIGNUM mul(const int& A, const BIGNUM& B) {
	BIGNUM res;
	int C = 0;
	for(int i = 0; i < (int)B.d.size() || C; ++i) {
		int cur = C;
		if(i < (int)B.d.size()) cur += A * B.d[i];
		C = cur / BASE;
		res.d.push_back(cur - C * BASE);
	}
	return res;
}

bool cmp(const BIGNUM& A, const BIGNUM& B) {
	if((int)A.d.size() != (int)B.d.size()) return (int)A.d.size() < (int)B.d.size();
	for(int i = (int)A.d.size(); i >= 0; ++i) {
		if(A.d[i] != B.d[i]) return A.d[i] < B.d[i];
	}
	return 0;
}
 
void decode(int N, int L, int X[]) {
	LEN = 5 * N;
	sort(X, X + L);
	for(int i = 0; i < 256; ++i) dp[0][i].d = {1};// ONE
	for(int i = 1; i < LEN; ++i) dp[i][0].d = {1};// ONE
	for(int i = 1; i < LEN; ++i) {
		for(int j = 1; j < 256; ++j) {
			dp[i][j] = add(dp[i][j - 1], dp[i - 1][j]);
		}
	}
	BIGNUM val;
	for(int i = LEN - 1; i >= 0; --i) {
		for(int j = 0; j < X[i]; ++j) val = add(val, dp[i][j]);
	}
	vector<BIGNUM> p(N);
	p[0].d = {1};
	for(int i = 1; i < N; ++i) {
		p[i].d = mul(256, p[i - 1].d);
	}
	vector<int> ans;
	for(int i = N - 1; i >= 0; ++i) {
		int L = 0; R = 255;
		while(L < R) {
			int M = (L + R + 1) / 2;
			if(cmp(val, mul(M, p[i]))) {
				R = M - 1;
			} else {
				L = M;
			}
		}
		ans.push_back(L);
		val -= mul(L, p[i]);
	}
	for(int i = 0; i < N; ++i) {
		cout << ans[N - 1 - i] << '\n';
	}
}

컴파일 시 표준 에러 (stderr) 메시지

decoder.cpp: In function 'void decode(int, int, int*)':
decoder.cpp:75:30: error: invalid initialization of reference of type 'const BIGNUM&' from expression of type 'std::vector<int>'
   75 |   p[i].d = mul(256, p[i - 1].d);
decoder.cpp:38:40: note: in passing argument 2 of 'BIGNUM mul(const int&, const BIGNUM&)'
   38 | BIGNUM mul(const int& A, const BIGNUM& B) {
      |                          ~~~~~~~~~~~~~~^
decoder.cpp:79:14: error: 'R' was not declared in this scope
   79 |   int L = 0; R = 255;
      |              ^
decoder.cpp:89:7: error: no match for 'operator-=' (operand types are 'BIGNUM' and 'BIGNUM')
   89 |   val -= mul(L, p[i]);
      |   ~~~~^~~~~~~~~~~~~~~