Submission #128989

# Submission time Handle Problem Language Result Execution time Memory
128989 2019-07-11T11:55:22 Z E869120 Permutation Recovery (info1cup17_permutation) C++14
10 / 100
45 ms 10108 KB
#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;
#pragma warning (disable: 4996)

const long long MAX_Z = 10000000000000000LL;

struct BigInteger {
	vector<long long> t;
};

bool operator==(BigInteger a1, BigInteger a2) {
	if (a1.t == a2.t) return true;
	return false;
}

bool operator<(BigInteger a1, BigInteger a2) {
	if (a1.t.size() < a2.t.size()) return true;
	if (a1.t.size() > a2.t.size()) return false;
	for (int i = (int)a1.t.size() - 1; i >= 0; i--) {
		if (a1.t[i] < a2.t[i]) return true;
		if (a1.t[i] > a2.t[i]) return false;
	}
	return false;
}

BigInteger operator+(BigInteger a1, BigInteger a2) {
	BigInteger a3; a3.t.resize(max(a1.t.size(), a2.t.size()) + 1, 0);

	for (int i = 0; i < a1.t.size(); i++) a3.t[i] += a1.t[i];
	for (int i = 0; i < a2.t.size(); i++) a3.t[i] += a2.t[i];
	for (int i = 0; i < (int)a3.t.size() - 1; i++) {
		while (a3.t[i] >= MAX_Z) { a3.t[i] -= MAX_Z; a3.t[i + 1] += 1; }
	}
	while (a3.t.size() >= 2 && a3.t[a3.t.size() - 1] == 0) a3.t.pop_back();
	return a3;
}

BigInteger operator-(BigInteger a1, BigInteger a2) {
	BigInteger a3; a3.t.resize(max(a1.t.size(), a2.t.size()) + 1, 0);

	for (int i = 0; i < a1.t.size(); i++) a3.t[i] += a1.t[i];
	for (int i = 0; i < a2.t.size(); i++) a3.t[i] -= a2.t[i];
	for (int i = 0; i < (int)a3.t.size() - 1; i++) {
		while (a3.t[i] < 0) { a3.t[i] += MAX_Z; a3.t[i + 1] -= 1; }
	}
	while (a3.t.size() >= 2 && a3.t[a3.t.size() - 1] == 0) a3.t.pop_back();
	return a3;
}

BigInteger ConvertFromString(string V) {
	BigInteger a1; a1.t.resize(((int)V.size() + 15) / 16);
	reverse(V.begin(), V.end());
	
	long long r = 1;
	for (int i = 0; i < V.size(); i++) {
		a1.t[i / 16] += r * (long long)(V[i] - '0');
		r *= 10;
		if (i % 16 == 15) r = 1;
	}
	return a1;
}

BigInteger getmin(BigInteger a1, BigInteger a2) {
	if (a1.t < a2.t) return a1;
	return a2;
}

pair<BigInteger, int> getmin(pair<BigInteger, int> a1, pair<BigInteger, int> a2) {
	if (a1.first < a2.first) return a1;
	return a2;
}

int N;
BigInteger P[100009]; int A[100009];

// -------------------- Segment Tree -------------------

int size_ = 0;
pair<BigInteger, int> dat[1 << 17]; BigInteger lazy[1 << 17]; BigInteger INF;

void init(int sz) {
	string str = ""; for (int i = 0; i < 400; i++) str += "9";
	INF = ConvertFromString(str);

	size_ = 1;
	while (size_ <= sz) size_ *= 2;

	for (int i = 1; i <= N; i++) dat[size_ + i] = make_pair(P[i], i);
	for (int i = size_ - 1; i >= 1; i--) dat[i] = getmin(dat[i * 2], dat[i * 2 + 1]);
}

void refresh(int pos) {
	if (pos == 10 && ConvertFromString("0") < lazy[21]) {
		pos += 0;
	}
	if (pos < size_) {
		lazy[pos * 2] = lazy[pos * 2] + lazy[pos];
		lazy[pos * 2 + 1] = lazy[pos * 2 + 1] + lazy[pos];
		dat[pos] = getmin(make_pair(dat[pos * 2].first - lazy[pos * 2], dat[pos * 2].second), make_pair(dat[pos * 2 + 1].first - lazy[pos * 2 + 1], dat[pos * 2 + 1].second));
		lazy[pos] = ConvertFromString("0");
	}
	else {
		dat[pos].first = dat[pos].first - lazy[pos];
		lazy[pos] = ConvertFromString("0");
	}
}

void lose_(int l, int r, BigInteger p, int a, int b, int u) {
	if (l <= a && b <= r) { lazy[u] = lazy[u] + p; refresh(u); return; }
	if (r <= a || b <= l) return;

	refresh(u);
	lose_(l, r, p, a, (a + b) >> 1, u * 2);
	lose_(l, r, p, (a + b) >> 1, b, u * 2 + 1);
	refresh(u);
}

void lose(int cl, int cr, BigInteger p) {
	lose_(cl, cr, p, 0, size_, 1);
}

pair<BigInteger, int> query_(int l, int r, int a, int b, int u) {
	refresh(u);
	if (l <= a && b <= r) return dat[u];
	if (r <= a || b <= l) return make_pair(INF, -1);
	
	pair<BigInteger, int> a1 = query_(l, r, a, (a + b) >> 1, u * 2);
	pair<BigInteger, int> a2 = query_(l, r, (a + b) >> 1, b, u * 2 + 1);

	if (a1.first < a2.first) return a1;
	return a2;
}

pair<BigInteger, int> query(int l, int r) {
	return query_(l, r, 0, size_, 1);
}

// -------------------- Segment Tree End ----------------

int main() {
	cin >> N;
	for (int i = 1; i <= N; i++) { string V; cin >> V; P[i] = ConvertFromString(V); }
	for (int i = N; i >= 1; i--) P[i] = (P[i] - P[i - 1]);
	
	init(N + 1);

	//for (int i = 1; i <= N; i++) cout << query(i, i + 1).first.t[0] << " "; cout << endl;

	int cnts = 0;
	while (cnts < N) {
		pair<BigInteger, int> c = query(1, N + 1);
		int pos = c.second;
		cnts++; A[pos] = cnts;
		lose(pos + 1, N + 1, P[pos]);
		//for (int i = 1; i <= N; i++) cout << query(i, i + 1).first.t[0] << " "; cout << endl;

		vector<int>pass; int cy = pos + size_;
		while (cy >= 1) { pass.push_back(cy); cy >>= 1; }
		reverse(pass.begin(), pass.end());
		for (int pos1 : pass) refresh(pos1);
		reverse(pass.begin(), pass.end());
		for (int pos1 : pass) refresh(pos1);

		dat[pos + size_] = make_pair(INF, -1);
		int cx = pos + size_;
		while (cx >= 2) {
			cx >>= 1;
			dat[cx] = min(dat[cx * 2], dat[cx * 2 + 1]);
		}
	}

	for (int i = 1; i <= N; i++) { if (i >= 2) cout << " "; cout << A[i]; } cout << endl;
	return 0;
}

Compilation message

permutation.cpp:6:0: warning: ignoring #pragma warning  [-Wunknown-pragmas]
 #pragma warning (disable: 4996)
 
permutation.cpp: In function 'BigInteger operator+(BigInteger, BigInteger)':
permutation.cpp:32:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < a1.t.size(); i++) a3.t[i] += a1.t[i];
                  ~~^~~~~~~~~~~~~
permutation.cpp:33:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < a2.t.size(); i++) a3.t[i] += a2.t[i];
                  ~~^~~~~~~~~~~~~
permutation.cpp: In function 'BigInteger operator-(BigInteger, BigInteger)':
permutation.cpp:44:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < a1.t.size(); i++) a3.t[i] += a1.t[i];
                  ~~^~~~~~~~~~~~~
permutation.cpp:45:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < a2.t.size(); i++) a3.t[i] -= a2.t[i];
                  ~~^~~~~~~~~~~~~
permutation.cpp: In function 'BigInteger ConvertFromString(std::__cxx11::string)':
permutation.cpp:58:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < V.size(); i++) {
                  ~~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Correct 16 ms 9848 KB Output is correct
2 Incorrect 45 ms 10108 KB Output isn't correct