Submission #448711

#	Time	Username	Problem	Language	Result	Execution time	Memory
448711		Antekb	Drvca (COCI19_drvca)	C++14	20 / 110	66 ms	3128 KiB

drvca

#include<bits/stdc++.h>
using namespace std;
bool czy(vector<int> V){
	//for(int i=0; i<V.size(); i++)cout<<V[i]<<" ";
	//	cout<<"\n";
	if(V.size()==0)return 0;
	if(V.size()==1)return 1;
	int d=V[1]-V[0];
	for(int i=1; i<V.size(); i++)if(V[i]-V[i-1]!=d)return 0;
	return 1;
}
void out(vector<int> V){
	cout<<V.size()<<"\n";
	for(int i:V)cout<<i<<" ";
		cout<<"\n";
}
vector<int> V;
bool roz(int a, int d){
	vector<int> A, B;
	for(int i:V){
		if(i==a)A.push_back(i), a+=d;
		else B.push_back(i);
	}
	if(B.size()==0){
		B.push_back(A.back());
		A.pop_back();
	}
	if(czy(B)){
		out(A);
		out(B);
		return 1;
	}
	d=0;
	for(int i=1; i<B.size(); i++){
		d=__gcd(B[i]-B[i-1], d);
	}
	if((V.back()-B[0])%d)return 0;
	reverse(V.begin(), V.end());
	d=-d;
	a=V[0];
	//cout<<a<<" "<<d<<"\n";
	A.clear();
	B.clear();
	for(int i:V){
		if(i==a)A.push_back(i), a+=d;
		else B.push_back(i);
	}
	reverse(A.begin(), A.end());
	reverse(B.begin(), B.end());
	if(B.size()==0){
		B.push_back(A.back());
		A.pop_back();
	}
	if(czy(B)){
		out(A);
		out(B);
		reverse(V.begin(), V.end());
		return 1;
	}
	B.push_back(A[0]);
	for(int i=0; i<A.size()-1; i++)A[i]=A[i+1];
	A.pop_back();
	sort(B.begin(), B.end());
	if(czy(B)){
		out(A);
		out(B);
		reverse(V.begin(), V.end());
		return 1;
	}
	reverse(V.begin(), V.end());
	return 0;
}
int main(){
	int n;
	cin>>n;
	V.resize(n);
	for(int &i:V)cin>>i;
	sort(V.begin(), V.end());
	if(roz(V[0], V[1]-V[0]))return 0;
	if(n==2){
		//cout<<1<<"\n"<<V[0]<<"\n"<<1<<"\n"<<V[1]<<"\n";
		return 0;
	}
	if(roz(V[0], V[2]-V[0]))return 0;
	if(roz(V[1], V[2]-V[1]))return 0;
	cout<<-1;
	return 0;
}

Compilation message (stderr)

drvca.cpp: In function 'bool czy(std::vector<int>)':
drvca.cpp:9:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |  for(int i=1; i<V.size(); i++)if(V[i]-V[i-1]!=d)return 0;
      |               ~^~~~~~~~~
drvca.cpp: In function 'bool roz(int, int)':
drvca.cpp:34:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |  for(int i=1; i<B.size(); i++){
      |               ~^~~~~~~~~
drvca.cpp:61:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |  for(int i=0; i<A.size()-1; i++)A[i]=A[i+1];
      |               ~^~~~~~~~~~~

• Subtask 1: 20 / 20
• Subtask 2: 0 / 30
• Subtask 3: 0 / 30
• Subtask 4: 0 / 30