hdu 4655 Cut Pieces（贪心）

题目链接：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;
}