QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#703454#9540. Double 11无限天堂 (Bohao Tu, Kaiming Qiu, Haozhou Li)#WA 1ms5924kbC++202.4kb2024-11-02 17:49:212024-11-02 17:49:22

Judging History

This is the latest submission verdict.

  • [2024-11-02 17:49:22]
  • Judged
  • Verdict: WA
  • Time: 1ms
  • Memory: 5924kb
  • [2024-11-02 17:49:21]
  • Submitted

answer

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

const int N = 2e5 + 5;
int n, m, s[N];
long long sum[N];
#define double long double
pair<double, int> dp[N];
pair<double, int> check(double k)
{
	for (int i = 1; i <= n; i++) dp[i] = {sqrtl((double)sum[i] * i) - k, 0};
	auto solve = [&](auto &solve, int l, int r) -> void
	{
		if (l == r) return;
		int mid = l + r >> 1;
		solve(solve, l, mid);
		for (int i = mid + 1, j = l; i <= r; i++)
		{
			while (j < mid)
			{
				auto tmp1 = dp[j - 1];
				tmp1.second += 1;
				tmp1.first += sqrtl((double)(sum[i] - sum[j - 1]) * (i - j + 1)) - k;
				auto tmp2 = dp[j];
				tmp2.second += 1;
				tmp2.first += sqrtl((double)(sum[i] - sum[j]) * (i - j)) - k;
				if (tmp2 < tmp1) j++;
				else break;
			}
			auto tmp = dp[j - 1];
			tmp.second += 1;
			tmp.first += sqrtl((double)(sum[i] - sum[j - 1]) * (i - j + 1)) - k;
			dp[i] = min(dp[i], tmp);
		}
		solve(solve, mid + 1, r);
	};
	solve(solve, 1, n);
	return dp[n];
}

//pair<double, int> check1(double k)
//{
//	for (int i = 1; i <= n; i++) dp[i] = {1e18, 0};
//	auto solve = [&](auto &solve, int l, int r, int a, int b) -> void
//	{
//		if (l > r) return;
//		if (a == b)
//		{
//			for (int i = l; i <= r; i++)
//			{	
//				auto tmp = dp[a - 1];
//				tmp.second += 1;
//				tmp.first += sqrtl((double)(sum[i] - sum[a - 1]) * (i - a + 1)) - k;
//				dp[i] = min(dp[i], tmp);
//			}
//			return;
//		}
//		int mid = l + r >> 1, p = a;
//		pair<double, int> mi{1e18, 0};
//		for (int i = a; i <= b and i <= mid; i++)
//		{
//			auto tmp = dp[i - 1];
//			tmp.second += 1;
//			tmp.first += sqrtl((double)(sum[mid] - sum[i - 1]) * (mid - i + 1)) - k;
//			if (tmp < mi) mi = tmp, p = i;
//		}
//		cout << l << ' ' << r << ' ' << a << ' ' << b << " : " << mid << ' ' << p << '\n';
//		dp[mid] = min(dp[mid], mi);
//		solve(solve, l, mid - 1, a, p);
//		solve(solve, mid + 1, r, p, b);
//	};
//	solve(solve, 1, n, 1, n);
//	return dp[n];
//}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	
	cin >> n >> m;
	for (int i = 1; i <= n; i++) cin >> s[i];
	sort(s + 1, s + n + 1);
	for (int i = 1; i <= n; i++) sum[i] = sum[i - 1] + s[i];
	
	double l = -1e9, r = 0, mid;
	int cnt = 80;
	while (cnt--)
	{
		mid = (l + r) / 2;
		if (check(mid).second >= m) r = mid;
		else l = mid;
	}
	double ans = check(r).first + r * m; 
	cout << fixed << setprecision(10) << ans << '\n';
} 

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3900kb

input:

4 2
1 2 3 4

output:

6.1911471296

result:

ok found '6.191147130', expected '6.191147130', error '0.000000000'

Test #2:

score: 0
Accepted
time: 0ms
memory: 3760kb

input:

10 3
1 2 3 4 5 6 7 8 9 10

output:

22.5916253665

result:

ok found '22.591625367', expected '22.591625367', error '0.000000000'

Test #3:

score: 0
Accepted
time: 1ms
memory: 5884kb

input:

1 1
1

output:

1.0000000000

result:

ok found '1.000000000', expected '1.000000000', error '0.000000000'

Test #4:

score: 0
Accepted
time: 1ms
memory: 5924kb

input:

1 1
100000

output:

316.2277660168

result:

ok found '316.227766017', expected '316.227766017', error '0.000000000'

Test #5:

score: 0
Accepted
time: 0ms
memory: 3820kb

input:

7 1
10 47 53 9 83 33 15

output:

41.8330013267

result:

ok found '41.833001327', expected '41.833001327', error '0.000000000'

Test #6:

score: -100
Wrong Answer
time: 0ms
memory: 3812kb

input:

8 5
138 1702 119 1931 418 1170 840 1794

output:

234.9338767076

result:

wrong answer 1st numbers differ - expected: '233.9016546', found: '234.9338767', error = '0.0044131'