제출 #779366

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
779366		Josia	식물 비교 (IOI20_plants)	C++17	0 / 100	4077 ms	12648 KiB

plants

#include "plants.h"

#include <bits/stdc++.h>

using namespace std;

int n, k;
vector<int> bigger, smaller;

vector<pair<int, int>> useToCalc;

int numgroups;



struct segtree {
	vector<int> tree;
	void init(int N) {
		tree.assign(N*4, 0);
	}
	void update(int v, int rl, int rr, int pos, int x) {
		if (rl == rr) {tree[v] = x; return;}
		int rm = (rl + rr)/2;

		if (pos<=rm) update(v*2, rl, rm, pos, x);
		else update(v*2+1, rm+1, rr, pos, x);
		tree[v] = tree[v*2] + tree[v*2+1];
	}
	int query(int v, int rl, int rr, int ql, int qr) {
		if (ql > qr) return 0;
		if (rl == ql && rr == qr) return tree[v];
		int rm = (rl + rr) / 2;
		return query(v*2, rl, rm, ql, min(qr, rm)) + query(v*2+1, rm+1, rr, max(ql, rm+1), qr);
	}
};



segtree blub1, blub2;








struct findExtrema {
	vector<int> findMax() {
		vector<int> mins;
		int last = 0;
		for (int i = 0; i<n; i++) {
			// cerr << last << "\n";
			if (bigger[i]-blub1.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub1.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0 && last >= k) {mins.push_back(i);}
			if (bigger[i]-blub1.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub1.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0) last = 0;
			last++;
		}
		for (int i = 0; i<n; i++) {
			// cerr << last << "\n";
			if (bigger[i]-blub1.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub1.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0 && last >= k) {mins.push_back(i);}
			if (bigger[i]-blub1.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub1.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0) last = 0;
			last++;
		}
		sort(mins.begin(), mins.end());
		mins.erase(unique(mins.begin(),mins.end()), mins.end());

		return mins;
	}

	vector<int> findMin() {
		vector<int> maxs;
		int last = 0;
		for (int i = 0; i<n; i++) {
			if (smaller[i]-blub2.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub2.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0 && last >= k) {maxs.push_back(i);}
			if (smaller[i]-blub2.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub2.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0) last = 0;
			last++;
		}
		for (int i = 0; i<n; i++) {
			if (smaller[i]-blub2.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub2.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0 && last >= k) {maxs.push_back(i);}
			if (smaller[i]-blub2.query(1, 0, n-1, i, min(n-1,i+k-1))-(i+k-1>=n?blub2.query(1, 0, n-1, 0, (i+k-1)%n):0) == 0) last = 0;
			last++;
		}
		sort(maxs.begin(), maxs.end());
		maxs.erase(unique(maxs.begin(),maxs.end()), maxs.end());

		return maxs;
	}
};



struct makeGroups {
	vector<pair<vector<int>, vector<int>>> groups;
	makeGroups(int x) {
		blub1.init(n);
		blub2.init(n);
		groups.clear();


		while (true) {
			// cerr << "---\n";
			// for (int i = 0; i<n; i++) {
			// 	cerr << bigger[i] << " " << smaller[i] << "\n";
			// }
			// cerr << "---\n";
			findExtrema machine;

			vector<int> mins = machine.findMin();
			vector<int> maxs = machine.findMax();

			if (mins.empty() && maxs.empty()) break;
			groups.push_back({mins, maxs});

			for (int i: maxs) {
				bigger[i] = 1e9;
				smaller[i] = 1e9;
				blub1.update(1, 0, n-1, i, 1);
				// for (int j = i-1; j>i-k; j--) {
				// 	if (bigger[(j+n)%n]!=1e9) bigger[(j+n)%n]--;
				// }
			}

			for (int i: mins) {
				bigger[i] = 1e9;
				smaller[i] = 1e9;
				blub2.update(1, 0, n-1, i, 1);
				// for (int j = i-1; j>i-k; j--) {
				// 	if (smaller[(j+n)%n]!=1e9) smaller[(j+n)%n]--;
				// }
			}
		}
	}	
};





void init(int K, std::vector<int> r) {
	n = r.size();
	k = K;
	smaller.assign(n,0);
	bigger = r;

	for (int i = 0; i<n; i++) {
		smaller[i] = k-1-bigger[i];
	}


	makeGroups groups(0);

	// for (auto i: groups.groups) {
	// 	for (auto j: i.first) {
	// 		cerr << j << " ";
	// 	}
	// 	cerr << "   ---   ";
	// 	for (auto j: i.second) {
	// 		cerr << j << " ";
	// 	}
	// 	cerr << "\n";
	// }

	// cerr << "initDone\n";


	vector<pair<int, int>> plants(n, {-1, -1});	// which group, 0min/1max/2both

	numgroups = groups.groups.size();

	for (int ind = 0; ind < groups.groups.size(); ind++) {
		auto i = groups.groups[ind];
		for (auto j: i.first) {
			plants[j] = {ind, 0};
		}
		for (auto j: i.second) {
			plants[j] = {ind, plants[j].second==-1?1:2};
		}
	}

	useToCalc = plants;

	return;
}

int compare_plants(int x, int y) {
	int g1 = useToCalc[x].first;
	int g2 = useToCalc[y].first;


	if (g1!=g2) {
		// if not connected return zero
		if (g1<g2) {
			if (useToCalc[x].second == 0) return -1;
			else return 1;
		}
		else {
			if (useToCalc[y].second == 0) return 1;
			else return -1;
		}
	} else {
		if (g1 < numgroups-1) {
			if (useToCalc[x].second == useToCalc[y].second) return 0;
			if (useToCalc[x].second < useToCalc[y].second) return -1;
			if (useToCalc[x].second > useToCalc[y].second) return 1;
		} else {
			if (abs(x-y)<k) {
				if (useToCalc[x].second == useToCalc[y].second) assert(0);
				if (useToCalc[x].second < useToCalc[y].second) return -1;
				if (useToCalc[x].second > useToCalc[y].second) return 1;
			} else {
				return 0;
			}
		}
	}


	return 0;
}

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

plants.cpp: In function 'void init(int, std::vector<int>)':
plants.cpp:170:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::vector<int>, std::vector<int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  170 |  for (int ind = 0; ind < groups.groups.size(); ind++) {
      |                    ~~~~^~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 14
• Subtask 3: 0 / 13
• Subtask 4: 0 / 17
• Subtask 5: 0 / 11
• Subtask 6: 0 / 15
• Subtask 7: 0 / 25