首页 > 代码库 > hdu 5009 Paint Pearls(dp)
hdu 5009 Paint Pearls(dp)
题目链接:hdu 5009 Paint Pearls
题目大意:给定一串珠子的目标颜色,现在要为这些珠子上色,每次可以选中一段区间上的珠子上色,代价为这段区间中颜色的数量k的平方,要求用最少的代价。
解题思路:dp[i]表示到i的最优代价,加上优化即可,当k(颜色总数)的平方大于N的可以直接跳出循环,当dp[i] > dp[i+1]时,可以考虑直接从dp[i+1]转移。
现场的时候交C++ TLE了,不然这题应该很快就出了,到最后交G++才过也是醉了。。。
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int maxn = 5 * 1e4 + 5;
const int INF = 0x3f3f3f3f;
struct point {
int val, pos;
point (int val = 0, int pos = 0) {
this->val = val;
this->pos = pos;
}
}p[maxn];
inline bool sort_val (const point& a, const point& b) {
return a.val < b.val;
}
inline bool sort_pos (const point& a, const point& b) {
return a.pos < b.pos;
}
int N, M, v[maxn];
void init () {
M = N;
for (int i = 1; i <= N; i++)
scanf("%d", &p[i].val);
int mv = 1;
for (int i = 2; i <= N; i++) {
if (p[i].val != p[i-1].val) {
p[++mv] = p[i];
p[mv].pos = mv;
}
}
N = mv;
sort(p + 1, p + 1 + N, sort_val);
mv = 0;
int pre = -1;
for (int i = 1; i <= N; i++) {
if (p[i].val != pre) {
pre = p[i].val;
mv++;
}
p[i].val = mv;
}
sort(p + 1, p + 1 + N, sort_pos);
}
int dp[maxn], vis[maxn];
int solve () {
if (N + 3 >= M)
return N;
vector<int> vec;
memset(dp, INF, sizeof(dp));
dp[0] = 0;
dp[N] = N;
int mv = 0;
while (true) {
int u = mv;
if (u == N)
return dp[u];
int cnt = 0, ans = dp[u + 1];
mv = u + 1;
for (int i = u + 1; i <= N; i++) {
if (vis[p[i].val] == 0) {
cnt++;
vis[p[i].val] = 1;
vec.push_back(p[i].val);
}
if (cnt * cnt + dp[u] > dp[N])
break;
if (dp[i] > dp[u] + cnt * cnt)
dp[i] = dp[u] + cnt * cnt;
if (dp[i] <= ans) {
ans = dp[i];
mv = i;
}
}
for (int i = 0; i < cnt; i++)
vis[vec[i]] = 0;
vec.clear();
}
return N;
}
int main () {
while (scanf("%d", &N) == 1) {
init();
printf("%d\n", solve());
}
return 0;
}
hdu 5009 Paint Pearls(dp)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。