原题(计蒜客 T-1224)
蒜头君想开家餐馆. 现在共有 nn 个地点可供选择。蒜头君打算从中选择合适的位置开设一些餐馆。这 nn 个地点排列在同一条直线上。我们用一个整数序列 m_1, m_2, ... m_nm1,m2,...mn 来表示他们的相对位置。由于地段关系, 开餐馆的利润会有所不同。我们用 p_ipi 表示在 m_imi 处开餐馆的利润。
为了避免自己的餐馆的内部竞争,餐馆之间的距离必须大于 kk。
请你帮助蒜头君选择一个总利润最大的方案。
输入格式
标准的输入包含若干组测试数据。
输入第一行是整数 T (1 \le T \le 1000)T(1≤T≤1000),表明有 TT 组测试数据。紧接着有 TT 组连续的测试。每组测试数据有 33 行;
第 11 行:地点总数 n (n < 100)n(n<100), 距离限制 k (k > 0k(k>0 && k < 1000)k<1000);
第 22 行: nn 个地点的位置 m_1 , m_2, ... m_n ( 1000000 > m_i > 0m1,m2,...mn(1000000>mi>0 且为整数, 升序排列);
第 33 行: nn 个地点的餐馆利润 p_1 , p_2, ... p_n ( 1000 > p_i > 0p1,p2,...pn(1000>pi>0 且为整数)。
输出格式
对于每组测试数据可能的最大利润
样例
输入:
2 3 11 1 2 15 10 2 30 3 16 1 2 15 10 2 30
输出:
40 30
思路
用a数组存餐馆位置,v数组存第 i 个餐馆的利润, 用 f 数组存从 i 到 j 餐馆的总利润,
代码
#include<iostream>
#include<bits/stdc++.h>
#include<string.h>
#define ll long long
using namespace std;
ll a[1000008];
ll v[1000008];
ll f[1000008];
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
ll N;
cin >> N;
while ( N-- )
{
ll n,k;
cin >> n >>k;
for(ll i=1;i<=n;i++)
cin >> a[i];
for(ll i=1;i<=n;i++)
{
cin >> v[i];
f[i]=v[i];
}
for(ll i=1;i<=n;i++)
{
for(ll j=1;j<=n;j++)
{
if(a[i]-a[j] > k )
{
f[i]=max(f[i],f[j]+v[i]);
}
}
}
ll maxx=0;
for(ll i=1;i<=n;i++)
maxx=max(maxx,f[i]);
cout<<maxx<<endl;
}
return 0;
}