KMIN - KMIN - Solution for SPOJ

Problem

https://vn.spoj.com/problems/KMIN

https://oj.vnoi.info/problem/KMIN

Cho 2 dãy số nguyên A và B. Với mọi số A[i]thuộc A và B[j] thuộc B người ta tính tổng nó. Tất cả các tổng này sau khi được sắp xếp không giảm sẽ tạo thành dãy C.

Nhiệm vụ của bạn là: Cho 2 dãy A, B. Tìm K số đầu tiên trong dãy C

Input

Dòng đầu tiên gồm 3 số: M, N, K

M dòng tiếp theo gồm M số mô tả dãy A

N dòng tiếp theo gồm N số mô tả dãy B

Output

Gồm K dòng tương ứng là K phần tử đầu tiên trong dãy C

Example

Input
4 4 6
1
2
3
4
2
3
4
5

Output
3
4
4
5
5
5

Giới hạn

1 ≤ M, N, K ≤ 50000
1 ≤ Ai, Bi ≤ 10^9

Tutorial

Submission

KMIN.cpp

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int,int> ii;
typedef unsigned long long ull;

#define X first
#define Y second
#define pb push_back
#define mp make_pair
#define ep emplace_back
#define EL printf("\n")
#define sz(A) (int) A.size()
#define FOR(i,l,r) for (int i=l;i<=r;i++)
#define FOD(i,r,l) for (int i=r;i>=l;i--)
#define fillchar(a,x) memset(a, x, sizeof (a))
#define faster ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);

const int N = 5e5+5;
int a[N], b[N], n;

struct data {
	int i, j, c;
	bool operator < (const data &o) const {
		return c > o.c;
	}
};

priority_queue<data> st;


int main() {
//  freopen("INP.TXT", "r", stdin);
//  freopen("OUT.TXT", "w", stdout);

	cin >> a[0] >> b[0] >> n;
	FOR(i,1,a[0]) scanf("%d", &a[i]);
	sort(a+1,a+a[0]+1);
	FOR(i,1,b[0]) scanf("%d", &b[i]);
	sort(b+1,b+b[0]+1);

	FOR(j,1,b[0]) st.push({1, j, a[1]+b[j]});

	FOR(i,1,n) {
		data t = st.top();
		st.pop();
		printf("%d\n", t.c);
		if (t.i < a[0])
			st.push({t.i+1, t.j, a[t.i+1] + b[t.j]});
	}


	return 0;
}

Tutorial SPOJ

KMIN - KMIN