hdu 4655 Cut Pieces（贪心）

来源：互联网发布：宁波seo入门教程编辑：程序博客网时间：2024/05/17 07:42

题目链接：hdu 4655 Cut Pieces

解题思路

同一块颜色的贡献值，我们算给最左边的位置。考虑相邻两块中后面一块的贡献值，a为i-1的颜色数量，b为i的颜色数量，除了这两块，其他位置的染色方案种数为c，那么位置i的贡献度为(max(a,b)−1)∗min(a,b))∗c

a > b：那么在如果i-1的为值染色为[b+1,a]的时候，i不管为什么颜色都可以增加一点贡献值；在[1,b]的时候，i则要避开i-1的颜色才能增加一点贡献，所以总的贡献度为(a−b)∗b+b∗(b−1)=(a−1)∗b
a < b：不管i-1的位置染什么颜色，i的位置都要避开这种颜色才能增一点贡献，所以a∗(b−1)

代码

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;typedef long long ll;const int maxn = 1e6 + 5;const int mod = 1e9 + 7;/* 相同块的贡献算给最左边*/int N, A[maxn], P[maxn];ll L[maxn], R[maxn];int solve () {    L[0] = R[N+1] = 1;    for (int i = 1; i <= N; i++) L[i] = L[i-1] * P[i] % mod;    for (int i = N; i >= 1; i--) R[i] = R[i+1] * P[i] % mod;    ll ret = R[1];    for (int i = 2; i <= N; i++) {        ll tmp = (1LL * max(P[i-1], P[i])-1) * min(P[i-1], P[i]) % mod;        ret = (ret + L[i-2] * R[i+1] % mod * tmp % mod) % mod;;    }    return ret;}int main () {    int cas;    scanf("%d", &cas);    while (cas--) {        scanf("%d", &N);        for (int i = 1; i <= N; i++) scanf("%d", &A[i]);        sort(A + 1, A + N + 1);        int p = N;        for (int i = 2; i <= N; i += 2) P[i] = A[p--];        for (int i = 1; i <= N; i += 2) P[i] = A[p--];        printf("%d\n", solve());    }    return 0;}

0 0