oj.uz

답안 #920037

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
920037 2024-02-02T02:13:51 Z Lobo 앵무새 (IOI11_parrots) C++17
81 / 100
 2000 ms 23380 KB 
#include "encoder.h"
#include "encoderlib.h"
#include <bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
#define fr first
#define sc second
#define int long long
// const int maxn = 1e5+10;
// const int inf = 1e18+10;
static int R;
static int k;
static int base = 1e8;

bool geq(vector<int> s, vector<int> t) {
	if(s.size() > t.size()) return 1;
	if(t.size() > s.size()) return 0;

	for(int i = (int) s.size() -1; i >= 0; i--) {
		if(s[i] > t[i]) return 1;
		if(t[i] > s[i]) return 0;
	}
	return 1;
}

vector<int> mult(vector<int> s, vector<int> t) {
	vector<int> ans((int) s.size()+(int) t.size(),0);
	if(s == vector<int>{0} or t == vector<int>{0}) return vector<int>{0};

	for(int i = 0; i < s.size(); i++) {
		for(int j = 0; j < t.size(); j++) {
			ans[i+j]+= s[i]*t[j];
		}
	}

	for(int i = 0; i < ans.size(); i++) {
		if(ans[i] >= base) {
			if(i+1 == (int) ans.size()) ans.pb(0);
			ans[i+1]+= ans[i]/base;
			ans[i]%= base;
		}
	}
	while(ans.size() > 1 && ans.back() == 0) ans.pop_back();
	return ans;
}

vector<int> add(vector<int> s, vector<int> t) {
	vector<int> ans = s;

	for(int i = 0; i < t.size(); i++) {
		if(i == (int) ans.size()) ans.pb(0);
		ans[i]+= t[i];
	}

	for(int i = 0; i < ans.size(); i++) {
		if(ans[i] >= base) {
			if(i+1 == (int) ans.size()) ans.pb(0);
			ans[i+1]+= ans[i]/base;
			ans[i]%= base;
		}
	}
	while(ans.size() > 1 && ans.back() == 0) ans.pop_back();
	return ans;
}

vector<int> trans(int x) {
	if(x == 0) return vector<int>{0};
	vector<int> ans;
	while(x != 0) {
		ans.pb(x%base);
		x/= base;
	}
	return ans;
}

int transback(vector<int> s) {
	int ans = 0;
	int p10 = 1;
	for(int i = 0; i < s.size(); i++) {
		ans+= p10*s[i];
		p10*= base;
	}
	return ans;
}

vector<int> dp[260][10*64+10];

void precomp() {
	// dp(i,sum) = # de (q_0,q_1,...,q_i) tal que a soma é no maximo sum
	for(int i = 0; i < R; i++) {
		for(int j = 0; j <= k; j++) {
			dp[i][j] = vector<int>{0};
		}
	}

	for(int j = 0; j <= k; j++) {
		dp[0][j] = trans(j+1);
	}

	for(int i = 1; i < R; i++) {
		dp[i][0] = trans(1);
		for(int j = 1; j <= k; j++) {
			dp[i][j] = add(dp[i][j],dp[i][j-1]);
			dp[i][j] = add(dp[i][j],dp[i-1][j]);
		}
	}
}

vector<int> querysmaller(vector<int> q) {
	vector<int> ans = {0};

	int sumq = 0;
	for(int i = R-1; i >= 0; i--) {
		for(int j = 0; j <= q[i]-1 && k-j-sumq >= 0; j++) {
			if(i == 0) ans = add(ans,trans((int)1));
			else ans = add(ans,dp[i-1][k-j-sumq]);
		}
		sumq+= q[i];
	}

	if(sumq <= k) ans = add(ans,trans(1));
	return ans;
}

vector<int> findg(vector<int> g) {
	vector<int> ans(R,0);

	int sumans = 0;
	vector<int> cur = {1};
	for(int i = R-1; i >= 0; i--) {
		ans[i] = 0;
		for(int j = 1; j <= k-sumans; j++) {
			vector<int> newcur;
			if(i == 0) newcur = add(cur,trans(1));
			else newcur = add(cur,dp[i-1][k-sumans-j+1]);
			if(!geq(g,newcur)) break;
			cur = newcur;
			ans[i] = j;
		}
		sumans+= ans[i];
	}
	return ans;
}

void encode(int32_t n, int32_t m[])
{
	k = 5*n;
	R = 210;
	precomp();

	vector<int> vm = trans(0);
	vector<int> p256 = trans(1);
	for(int i = 0; i < n; i++) {
		vm = add(vm,mult(p256,trans(m[i])));
		p256 = mult(p256,trans(256));
	}

	vector<int> ans = findg(add(vm,trans(1)));

	for(int i = 0; i < ans.size(); i++) {
		for(int j = 0; j < ans[i]; j++) {
			send((int32_t) i);
		}
	}	
}

#include "decoder.h"
#include "decoderlib.h"
#include <bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
#define fr first
#define sc second
#define int long long
// const int maxn = 1e5+10;
// const int inf = 1e18+10;
static int R1;
static int k1;
static int base1 = 1e8;
bool geq1(vector<int> s, vector<int> t) {
	if(s.size() > t.size()) return 1;
	if(t.size() > s.size()) return 0;

	for(int i = (int) s.size() -1; i >= 0; i--) {
		if(s[i] > t[i]) return 1;
		if(t[i] > s[i]) return 0;
	}
	return 1;
}

vector<int> mult1(vector<int> s, vector<int> t) {
	vector<int> ans((int) s.size()+(int) t.size(),0);
	if(s == vector<int>{0} or t == vector<int>{0}) return vector<int>{0};

	for(int i = 0; i < s.size(); i++) {
		for(int j = 0; j < t.size(); j++) {
			ans[i+j]+= s[i]*t[j];
		}
	}

	for(int i = 0; i < ans.size(); i++) {
		if(ans[i] >= base1) {
			if(i+1 == (int) ans.size()) ans.pb(0);
			ans[i+1]+= ans[i]/base1;
			ans[i]%= base1;
		}
	}
	while(ans.size() > 1 && ans.back() == 0) ans.pop_back();
	return ans;
}

vector<int> neg(vector<int> s) {
	for(int i = 0; i < s.size(); i++) {
		s[i]*= -1;
	}
	return s;
}

vector<int> add1(vector<int> s, vector<int> t) {
	vector<int> ans = s;

	for(int i = 0; i < t.size(); i++) {
		if(i == (int) ans.size()) ans.pb(0);
		ans[i]+= t[i];
	}

	for(int i = 0; i < ans.size(); i++) {

		if(ans[i] < 0) {
			if(i+1 == (int) ans.size()) ans.pb(0);
			int qt = (ans[i]-1)/base1 + 1;
			ans[i]+= qt*base1;
			ans[i+1]-= qt;
		}
		if(ans[i] >= base1) {
			if(i+1 == (int) ans.size()) ans.pb(0);
			ans[i+1]+= ans[i]/base1;
			ans[i]%= base1;
		}
	}
	while(ans.size() > 1 && ans.back() == 0) ans.pop_back();

	return ans;
}

vector<int> trans1(int x) {
	if(x == 0) return vector<int>{0};
	vector<int> ans;
	while(x != 0) {
		ans.pb(x%base1);
		x/= base1;
	}
	return ans;
}

int trans1back1(vector<int> s) {
	int ans = 0;
	int p10 = 1;
	for(int i = 0; i < s.size(); i++) {
		ans+= p10*s[i];
		p10*= base1;
	}
	return ans;
}

vector<int> dp1[260][10*64+10];

void precomp1() {
	// dp(i,sum) = # de (q_0,q_1,...,q_i) tal que a soma é no maximo sum
	for(int i = 0; i < R1; i++) {
		for(int j = 0; j <= k1; j++) {
			dp1[i][j] = vector<int>{0};
		}
	}

	for(int j = 0; j <= k1; j++) {
		dp1[0][j] = trans1(j+1);
	}

	for(int i = 1; i < R1; i++) {
		dp1[i][0] = trans1(1);
		for(int j = 1; j <= k1; j++) {
			dp1[i][j] = add1(dp1[i][j],dp1[i][j-1]);
			dp1[i][j] = add1(dp1[i][j],dp1[i-1][j]);
		}
	}
}

vector<int> querysmaller1(vector<int> q) {
	vector<int> ans = {0};

	int sumq = 0;
	for(int i = R1-1; i >= 0; i--) {
		for(int j = 0; j <= q[i]-1 && k1-j-sumq >= 0; j++) {
			if(i == 0) ans = add1(ans,trans1((int)1));
			else ans = add1(ans,dp1[i-1][k1-j-sumq]);
		}
		sumq+= q[i];
	}

	if(sumq <= k1) ans = add1(ans,trans1(1));
	return ans;
}

vector<int> findg1(vector<int> g) {
	vector<int> ans(R1,0);

	int sumans = 0;
	for(int i = R1-1; i >= 0; i--) {
		ans[i] = 0;
		for(int j = 1; j+sumans <= k1; j++) {
			ans[i] = j;
			if(!(geq1(g,querysmaller1(ans)))) {
				ans[i] = j-1;
				break;
			}
		}
		sumans+= ans[i];
	}
	return ans;
}

void decode(int32_t n, int32_t L, int32_t X[])
{
	k1 = 5*n;
	R1 = 210;
	precomp1();
	vector<int> q(R1,0);
	for(int i = 0; i < L; i++) {
		q[X[i]]++;
	}
	vector<int> vm = add1(querysmaller1(q),neg(trans1(1)));

	vector<int> m(n);
	for(int i = n-1; i >= 0; i--) {
		vector<int> p256 = trans1(1);
		for(int j = 0; j < i; j++) {
			p256 = mult1(p256,trans1(256));
		}

		m[i] = 0;
		while(geq1(vm,p256)) {
			m[i]++;
			vm = add1(vm,neg(p256));
		}
	}

	for(int i =0; i < n; i++) {
		output((int32_t) m[i]);
	}

}

Compilation message

encoder.cpp: In function 'std::vector<long long int> mult(std::vector<long long int>, std::vector<long long int>)':
encoder.cpp:31:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~
encoder.cpp:32:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |   for(int j = 0; j < t.size(); j++) {
      |                  ~~^~~~~~~~~~
encoder.cpp:37:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |  for(int i = 0; i < ans.size(); i++) {
      |                 ~~^~~~~~~~~~~~
encoder.cpp: In function 'std::vector<long long int> add(std::vector<long long int>, std::vector<long long int>)':
encoder.cpp:51:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   51 |  for(int i = 0; i < t.size(); i++) {
      |                 ~~^~~~~~~~~~
encoder.cpp:56:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |  for(int i = 0; i < ans.size(); i++) {
      |                 ~~^~~~~~~~~~~~
encoder.cpp: In function 'long long int transback(std::vector<long long int>)':
encoder.cpp:80:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   80 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~
encoder.cpp: In function 'void encode(int32_t, int32_t*)':
encoder.cpp:161:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  161 |  for(int i = 0; i < ans.size(); i++) {
      |                 ~~^~~~~~~~~~~~

decoder.cpp: In function 'std::vector<long long int> mult1(std::vector<long long int>, std::vector<long long int>)':
decoder.cpp:30:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~
decoder.cpp:31:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |   for(int j = 0; j < t.size(); j++) {
      |                  ~~^~~~~~~~~~
decoder.cpp:36:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |  for(int i = 0; i < ans.size(); i++) {
      |                 ~~^~~~~~~~~~~~
decoder.cpp: In function 'std::vector<long long int> neg(std::vector<long long int>)':
decoder.cpp:48:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~
decoder.cpp: In function 'std::vector<long long int> add1(std::vector<long long int>, std::vector<long long int>)':
decoder.cpp:57:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |  for(int i = 0; i < t.size(); i++) {
      |                 ~~^~~~~~~~~~
decoder.cpp:62:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |  for(int i = 0; i < ans.size(); i++) {
      |                 ~~^~~~~~~~~~~~
decoder.cpp: In function 'long long int trans1back1(std::vector<long long int>)':
decoder.cpp:94:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   94 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~

Subtask #1
17.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 38 ms 11452 KB Output is correct

Subtask #2
17.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 183 ms 11820 KB Output is correct
2 Correct 263 ms 12240 KB Output is correct
3 Correct 388 ms 12996 KB Output is correct
4 Correct 429 ms 13016 KB Output is correct

Subtask #3
18.0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 180 ms 11584 KB Output is correct
2 Correct 290 ms 12104 KB Output is correct
3 Correct 380 ms 12984 KB Output is correct
4 Correct 420 ms 12900 KB Output is correct

Subtask #4
29.0 / 29.0

# 결과 실행 시간 메모리 Grader output
1 Correct 177 ms 11468 KB Output is correct
2 Correct 423 ms 12988 KB Output is correct
3 Correct 558 ms 14272 KB Output is correct
4 Correct 929 ms 15832 KB Output is correct
5 Correct 975 ms 16204 KB Output is correct
6 Correct 1016 ms 16104 KB Output is correct

Subtask #5
0.0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 428 ms 13020 KB Output is correct - P = 5.000000
2 Correct 1010 ms 16068 KB Output is correct - P = 5.000000
3 Correct 1051 ms 16324 KB Output is correct - P = 5.000000
4 Correct 1759 ms 20592 KB Output is correct - P = 5.000000
5 Execution timed out 2105 ms 22672 KB Time limit exceeded
6 Execution timed out 2151 ms 23380 KB Time limit exceeded
7 Incorrect 1174 ms 22608 KB Error : Output is wrong