题目:https://nanti.jisuanke.com/t/90
近期,法师康采纳了员工们的提议,计划在总部大楼内部种植园艺,以提供更加温馨的工作环境。公司将园艺设计的任务交给了晓萌同学。
公司总部大楼内部的构造可以分为n个区域,编号为0, 1, …, n–1,其中区域i与i + 1是相邻的(0 ≤ i < n – 1)。根据员工的投票和反馈,晓萌拿到了一份数据,表明在区域i种植园艺可以获得员工的满意度为Ai。晓萌希望园艺的布置方案满足条件:1.至少覆盖m个区域;2.布置园艺的区域是连续的。请帮他找到一种满足条件的方案,使布置园艺区域的员工的满意度的平均值最大。
输入第1行包含两个整数n和m,分别表示总区域数和至少覆盖的区域数。第2行包含n个整数A0, A1,…, An–1,依次表示在每个区域种植园艺可以获得员工的满意度。(1 ≤ m ≤ n ≤ 106,1 ≤ Ai ≤ 106)
输出包括一行,表示员工的平均满意度的最大值。如果这个数是一个整数,则直接按整数格式输出;否则,请用最简分数表示,分子分母以“/”分割,格式见样例。
题目大意:
这是一道最大化平均值的问题。先看一下常规的最大化平均值问题。
Yougth现在有n个物品的重量和价值分别是Wi和Vi,你能帮他从中选出k个物品使得单位重量的价值最大吗?
样例输入
3 2
2 2
5 3
2 1
样例输出 选择2 2 和2 1。(2+1)/(2+2)
0.75
一般的想法是把物品的单位价值排序,然后贪心的选取,但是是错误的。因为选取2 2 和5 3结果是错误的。因为平均值1和0.6与1和0.5,不是简单的(1+0.5)/2,还有两个数还有比重。
另C(X):=选择使得单位重量的价值不小于x。s是某一集合。原问题变成求解最大的x。
然后贪心选取前K,看能不能满足。
接下来就是二分查找x的值,直到找到最大的x。
然后回到本题,这道题大致意思就是n个连续的数,从中选出m个连续的数,使得单位价值最大。
其实就是从不连续的数改成连续的数。就成了最大子段和问题,但是这个最大子段和限定了子段长度。一般子段和问题复杂度为
O(N),现在限定了子段长度大于K。
#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
#define maxn 1000000+5
long long a[maxn];
double b[maxn],sum[maxn];
double dp[maxn];//求最大子段和
#define INF 1<<29
long long start, maxnum;
bool judge(double x,long long n,long long k)
{
for (int i = 1; i <= n; i++)
{
b[i] = a[i] - x ;//代价为1,变
sum[i] = sum[i-1]+b[i];
if (i == 1)
dp[i] = sum[1];
else
dp[i] = min(sum[i], dp[i - 1]);//前i项中最小的sum[i]
}
double ans = -INF;
maxnum = 0;
for (int i = k; i <= n; i++)//寻找长度大于K的子段。
{
if (ans < sum[i] - dp[i - k])
{
ans = sum[i] - dp[i - k];
maxnum = i; //子段的尾巴
}
}
double s = 0;
for (int i = maxnum; i > 0; i--)
{
s += b[i];
if (s == ans)
{
start = i;
break;
}
}
return ans >= 0;
}
long long cal(long long a, long long b)
{
return b ? cal(b, a%b) : a;
}
void f(long long m,long long n)
{
long long fenzi=0, fenmu=0;
double ans=0,left=0,right=10000000,mid;
while (right - left > 0.000001)
{
mid = (left + right) / 2;
if (judge(mid, n, m))
left = mid;
else
right = mid;
}
for (long long i = start; i <= maxnum; i++)
fenzi += a[i];
fenmu = maxnum - start + 1;
long long yue = cal(fenzi, fenmu);
if (fenzi%fenmu == 0)
cout << fenzi/fenmu;
else
cout << fenzi / yue<<"/" << fenmu / yue;
}
int main()
{
long long n, m, x;
scanf("%lld%lld", &n,&m);
for (long long i = 1; i <= n; i++)
scanf("%lld", &a[i]);
f(m, n);
return 0;
}