oj.uz

Submission #796093

# Submission time Handle Problem Language Result Execution time Memory
796093 2023-07-28T06:12:20 Z ono_de206 Parrots (IOI11_parrots) C++14
100 / 100
 1658 ms 27176 KB 
#include "encoder.h"
#include "encoderlib.h"
#include<bits/stdc++.h>
using namespace std;

#define in insert
#define all(x) x.begin(),x.end()
#define pb push_back
#define eb emplace_back
#define ff first
#define ss second

//#define int long long
 
typedef long long ll;
typedef vector<int> vi;
typedef set<int> si;
typedef multiset<int> msi;
typedef pair<int, int> pii;
typedef vector<pii> vpii;

const int B = 10000, A = 40;
struct bigNum {
	array<int, A> dig;
	bigNum(int x = 0) {
		for(int i = 0; i < A; i++) {
			dig[i] = x % B;
			x /= B;
		}
	}
	bigNum& operator+=(const bigNum &b) {
		int carry = 0;
		for(int i = 0; i < A; i++) {
			dig[i] += b.dig[i] + carry;
			carry = dig[i] / B;
			dig[i] %= B;
		}
		return *this;
	};
	bigNum& operator-=(const bigNum &b) {
		for(int i = 0; i < A; i++) {
			dig[i] -= b.dig[i];
			while(dig[i] < 0) {
				assert(i + 1 < A);
				dig[i] += B;
				dig[i + 1]--;
			}
		}
		return *this;
	}
	bigNum& operator*=(const int &b) {
		int carry = 0;
		for(int i = 0; i < A; i++) {
			dig[i] *= b;
			dig[i] += carry;
			carry = dig[i] / B;
			dig[i] %= B;
		}
		return *this;
	}
	friend bigNum operator+(const bigNum &a, const bigNum &b) {
		return bigNum(a) += b;
	}
	friend bigNum operator-(const bigNum &a, const bigNum &b) {
		return bigNum(a) -= b;
	}
	friend bigNum operator*(const bigNum &a, const int &b) {
		return bigNum(a) *= b;
	}
	int compare(const bigNum &a) {
		for(int i = A - 1; i >= 0; i--) {
			if(a.dig[i] > dig[i]) return -1;
			else if(a.dig[i] < dig[i]) return 1;
		}
		return 0;
	}
	bool operator<(const bigNum &b) {
		return this->compare(b) == -1;
	}
	bool operator>(const bigNum &b) {
		return this->compare(b) == 1;
	}
	bool operator==(const bigNum &b) {
		return this->compare(b) == 0;
	}
	bool operator!=(const bigNum &b) {
		return this->compare(b) != 0;
	}
	friend ostream& operator<<(ostream& out, const bigNum &a) {
		for(int i = 0; i < 10; i++) {
			out << a.dig[i];
			if(i + 1 < A) out << ", ";
		}
		out << ';';
	}
};


void encode(int n, int a[]) {
	bigNum val(0);
	vector<bigNum> pw(1, 1);
	for(int i = 1; i < n; i++) {
		pw.pb(pw.back() * 256);
	}
	for(int i = 0; i < n; i++) {
		val += pw[i] * a[i];
	}
	int len = n * 5;
	vector<vector<bigNum>> dp(len + 1, vector<bigNum>(256, 0));
	// dp[0] = vector<bigNum>(256, 1);
	dp[0][0] = 1;
	for(int i = 1; i <= len; i++) {
		for(int j = 0; j < 256; j++) {
			dp[i][j] = dp[i - 1][j];
			if(j - 1 >= 0) dp[i][j] += dp[i][j - 1];
		}
	}
	for(int i = len; i >= 1; i--) {
		int ans = 0;
		for(int j = 0; j < 256; j++) {
			if(dp[i][j] > val) {
				ans = j;
				break;
			}
			val -= dp[i][j];
		}
		send(ans);
	}
}
	
#include "decoder.h"
#include "decoderlib.h"
#include<bits/stdc++.h>
using namespace std;

#define in insert
#define all(x) x.begin(),x.end()
#define pb push_back
#define eb emplace_back
#define ff first
#define ss second

//#define int long long
 
typedef long long ll;
typedef vector<int> vi;
typedef set<int> si;
typedef multiset<int> msi;
typedef pair<int, int> pii;
typedef vector<pii> vpii;

const int B = 10000, A = 40;
struct bigNum {
	array<int, A> dig;
	bigNum(int x = 0) {
		for(int i = 0; i < A; i++) {
			dig[i] = x % B;
			x /= B;
		}
	}
	bigNum& operator+=(const bigNum &b) {
		int carry = 0;
		for(int i = 0; i < A; i++) {
			dig[i] += b.dig[i] + carry;
			carry = dig[i] / B;
			dig[i] %= B;
		}
		return *this;
	};
	bigNum& operator-=(const bigNum &b) {
		for(int i = 0; i < A; i++) {
			dig[i] -= b.dig[i];
			while(dig[i] < 0) {
				dig[i] += B;
				dig[i + 1]--;
			}
		}
		return *this;
	}
	bigNum& operator*=(const int &b) {
		int carry = 0;
		for(int i = 0; i < A; i++) {
			dig[i] = dig[i] * b + carry;
			carry = dig[i] / B;
			dig[i] %= B;
		}
		return *this;
	}
	friend bigNum operator+(const bigNum &a, const bigNum &b) {
		return bigNum(a) += b;
	}
	friend bigNum operator-(const bigNum &a, const bigNum &b) {
		return bigNum(a) -= b;
	}
	friend bigNum operator*(const bigNum &a, const int &b) {
		return bigNum(a) *= b;
	}
	int compare(const bigNum &a) {
		for(int i = A - 1; i >= 0; i--) {
			if(a.dig[i] > dig[i]) return -1;
			else if(a.dig[i] < dig[i]) return 1;
		}
		return 0;
	}
	bool operator<(const bigNum &b) {
		return this->compare(b) == -1;
	}
	bool operator>(const bigNum &b) {
		return this->compare(b) == 1;
	}
	bool operator==(const bigNum &b) {
		return this->compare(b) == 0;
	}
	bool operator!=(const bigNum &b) {
		return this->compare(b) != 0;
	}
	friend ostream& operator<<(ostream& out, const bigNum &a) {
		for(int i = 0; i < 10; i++) {
			out << a.dig[i];
			if(i + 1 < A) out << ", ";
		}
		out << ';';
	}
};

void decode(int n, int l, int x[]) {
	sort(x, x + l);
	int len = n * 5;
	vector<vector<bigNum>> dp(len + 1, vector<bigNum>(256, 0));
	// dp[0] = vector<bigNum>(256, 1);
	dp[0][0] = 1;
	for(int i = 1; i <= len; i++) {
		for(int j = 0; j < 256; j++) {
			dp[i][j] = dp[i - 1][j];
			if(j - 1 >= 0) dp[i][j] += dp[i][j - 1];
		}
	}
	vector<int> ret;
	bigNum val(0);
	for(int i = l - 1; i >= 0; i--) {
		for(int j = 0; j < x[i]; j++) {
			val += dp[i + 1][j];
		}
	}
	// cout << val << " val here\n";
	vector<bigNum> pw(1, 1);
	for(int i = 1; i < n; i++) {
		pw.pb(pw.back() * 256);
	}
	// for(int i = 0; i < n; i++) {
	// 	cout << pw[i] << '\n';
	// }
	// cout << '\n';
	for(int i = n - 1; i >= 0; i--) {
		// cout << val << '\n';
		int ans = 0;
		while(pw[i] < val || pw[i] == val) {
			ans++;
			val -= pw[i];
		}
		ret.pb(ans);
	}
	reverse(all(ret));
	for(int x : ret) {
		// cout << x << ' ';
		output(x);
	}
	// cout << '\n';
}

Compilation message

encoder.cpp: In function 'std::ostream& operator<<(std::ostream&, const bigNum&)':
encoder.cpp:95:2: warning: no return statement in function returning non-void [-Wreturn-type]
   95 |  }
      |  ^

decoder.cpp: In function 'std::ostream& operator<<(std::ostream&, const bigNum&)':
decoder.cpp:93:2: warning: no return statement in function returning non-void [-Wreturn-type]
   93 |  }
      |  ^

Subtask #1
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 30 ms 4136 KB Output is correct

Subtask #2
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 148 ms 4560 KB Output is correct
2 Correct 213 ms 5652 KB Output is correct
3 Correct 293 ms 7460 KB Output is correct
4 Correct 307 ms 7760 KB Output is correct

Subtask #3
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 154 ms 4436 KB Output is correct
2 Correct 204 ms 5688 KB Output is correct
3 Correct 285 ms 7280 KB Output is correct
4 Correct 322 ms 7716 KB Output is correct

Subtask #4
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 148 ms 4464 KB Output is correct
2 Correct 312 ms 7996 KB Output is correct
3 Correct 394 ms 9336 KB Output is correct
4 Correct 614 ms 13456 KB Output is correct
5 Correct 666 ms 13848 KB Output is correct
6 Correct 667 ms 14236 KB Output is correct

Subtask #5
19.0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 313 ms 7684 KB Output is correct - P = 5.000000
2 Correct 679 ms 14368 KB Output is correct - P = 5.000000
3 Correct 685 ms 14852 KB Output is correct - P = 5.000000
4 Correct 1190 ms 21472 KB Output is correct - P = 5.000000
5 Correct 1495 ms 25700 KB Output is correct - P = 5.000000
6 Correct 1603 ms 26856 KB Output is correct - P = 5.000000
7 Correct 1658 ms 27176 KB Output is correct - P = 5.000000