Submission #746688

#	Time	Username	Problem	Language	Result	Execution time	Memory
746688		Abrar_Al_Samit	Table Tennis (info1cup20_tabletennis)	C++17	100 / 100	333 ms	33824 KiB

tabletennis

#include<bits/stdc++.h>
using namespace std;

const int nax = 150402;

int a[nax];
int n, k;

void check(int sum) {
	int l = 1, r = n+k;
	vector<int>b;
	int cur = a[1] + a[n+k];
	while(l<r) {
		if(cur>sum) {
			cur -= a[r];
			--r;
			cur += a[r];
		}
		else if(cur<sum) {
			cur -= a[l];
			++l;
			cur += a[l];
		} else {
			b.push_back(a[l]);
			b.push_back(a[r]);

			if(b.size()==n) {
				sort(b.begin(), b.end());
				for(int x : b) {
					cout<<x<<' ';
				}
				cout<<'\n';
				exit(0);
			}

			++l, --r;
			cur = a[l] + a[r];
		}
	}
}
void PlayGround() {
	cin>>n>>k;
	for(int i=1; i<=n+k; ++i) {
		cin>>a[i];
	}

	if(n+k < 4 * k) {
		for(int i=1; i<=n+k; ++i) {
			for(int j=i+1; j<=n+k; ++j) {
				check(a[i]+a[j]);
			}
		}
	} else {
		map<int, int>cnt;
		for(int i=1; i<=2*k; ++i) {
			for(int j=n-k+1; j<=n+k; ++j) {
				cnt[a[i]+a[j]]++;
			}
		}
		for(auto [x, y] : cnt) {
			if(y >= k) {
				check(x);
			}
		}
	}

	// cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
}
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	PlayGround();
	return 0;
}

Compilation message (stderr)

tabletennis.cpp: In function 'void check(int)':
tabletennis.cpp:27:15: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   27 |    if(b.size()==n) {
      |       ~~~~~~~~^~~

• Subtask 1: 11 / 11
• Subtask 2: 9 / 9
• Subtask 3: 14 / 14
• Subtask 4: 15 / 15
• Subtask 5: 9 / 9
• Subtask 6: 14 / 14
• Subtask 7: 15 / 15
• Subtask 8: 13 / 13