首页 > 代码库 > hdu 4719 Oh My Holy FFF(线段数+dp)
hdu 4719 Oh My Holy FFF(线段数+dp)
题目链接:hdu 4719 Oh My Holy FFF
题目大意:队伍里有n个人,给出每个人的身高,他们按照顺序排列,现在要将这n个人分成若干组,每一组的人数不得大于l,并且第i组的最后一个人的身高一定要大于第i?1组的最后一个人的身高。要求最后的权值最大,权值计算方法在题目中,k为组号。
解题思路:dp[i]表示以第i个人作为结尾的最大权值,那么dp[i]肯定是从前面的l-1个中转移过来的,即dp[i]=dp[j]+h[i]2?h[j] 要求h[i]>h[j].
但是这样的复杂度为o(n2),然后n最大为105,时间上不能接受,所以用线段树代替查询操作,但是转移的条件有说h[i]>h[j],所以我们要先将每个人按照身高排序,这样就逐个计算就不需要考虑身高的限制,因为如果已经被更新了的值,身高肯定小于当前需要考虑的人。
并且将要查找的值b[i]=dp[i]?h[j].
线段树没正式接触过,所以写的非常搓,也没有延时更新。
代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e5+5;
struct state {
int pos;
ll high;
}p[N];
struct node {
int left;
int right;
ll val;
}t[N*4];
int n, k;
ll dp[N];
inline ll max(ll a, ll b) {
return a > b ? a : b;
}
ll BuildTree (int c, int l, int r) {
t[c].left = l;
t[c].right = r;
if (l == r) {
t[c].val = -1;
} else {
int mid = (l + r)/2;
t[c].val = max(BuildTree(c*2, l, mid), BuildTree(c*2+1, mid+1, r));
}
return t[c].val;
}
ll Query (int c, int l, int r) {
if (l == t[c].left && r == t[c].right)
return t[c].val;
int mid = (t[c].left + t[c].right) / 2;
if (l <= mid && r > mid)
return max(Query(c*2, l, mid), Query(c*2+1, mid+1, r));
else if (l <= mid && r <= mid)
return Query(c*2, l, r);
else
return Query(c*2+1, l, r);
}
ll upDate (int c, int l, int r, ll val) {
if (l == t[c].left && r == t[c].right) {
t[c].val = val;
return val;
}
int mid = (t[c].left + t[c].right) / 2;
if (l <= mid && r > mid)
return t[c].val = max(upDate(c*2, l, mid, val), upDate(c*2+1, mid+1, r, val));
else if (l <= mid && r <= mid)
return t[c].val = max(upDate(c*2, l, r, val), t[c*2+1].val);
else
return t[c].val = max(t[c*2].val, upDate(c*2+1, l, r, val));
}
inline bool cmp (const state& a, const state& b) {
if (a.high != b.high)
return a.high < b.high;
return a.pos > b.pos;
}
void init () {
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i++) {
cin >> p[i].high;
p[i].pos = i;
}
sort(p+1, p+n+1, cmp);
BuildTree(1, 0, n);
}
ll solve () {
upDate(1, 0, 0, 0);
for (int i = 1; i <= n; i++) {
dp[p[i].pos] = -1;
ll val = Query(1, max(0, p[i].pos-k), p[i].pos-1);
//cout << val << " " << p[i].pos << endl;
if (val == -1)
continue;
dp[p[i].pos] = val + p[i].high * p[i].high;
if (p[i].pos == n)
break;
upDate(1, p[i].pos, p[i].pos, dp[p[i].pos] - p[i].high);
}
return dp[n];
}
int main () {
int cas;
scanf("%d", &cas);
for (int i = 1; i <= cas; i++) {
init ();
cout << "Case #" << i << ": ";
ll ans = solve();
if (ans <= 0)
cout << "No solution" << endl;
else
cout << ans << endl;
}
return 0;
}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。