【题解】P2267 琪琪的项链

最新推荐文章于 2024-08-21 18:16:36 发布

inferior_hjx

最新推荐文章于 2024-08-21 18:16:36 发布

阅读量373

点赞数 3

分类专栏：题解文章标签： c++ 算法

本文链接：https://blog.csdn.net/inferior_hjx/article/details/130547059

版权

题解专栏收录该内容

32 篇文章 0 订阅

订阅专栏

文章介绍了一道关于计算不同颜色珠子序列方案数的编程题，采用动态规划方法解决。通过对珠子颜色离散化，建立状态转移方程，计算以特定颜色珠子结尾的序列数。通过预处理和前缀和优化，实现了O(nlogn)的时间复杂度解决方案。

摘要由CSDN通过智能技术生成

link

题目大意

$n(1\le n\le 5\times 10^5)$ 颗带颜色的珠子，从左到右选出若干颗珠子（不少于 $1$ 颗），问有多少种方案。答案模 $m$ .

题解

小 dp 题。

首先对珠子颜色离散化，记得开 long long. 不多说。

观察发现，选出珠子不考虑原数组中的位置，只需要保证珠子序列在原数组中是从左到右的。

故设状态 $f [i]$ 表示以 $a [i]$ 结尾的珠子序列的方案数。

从左到右枚举，考虑第 $i$ 颗珠子（颜色为 $a [i]$ ）能接在哪些序列后。记离第 $i$ 颗珠子最近的颜色与这颗珠子相同的珠子位置为 $l s t [i]$ . 那么有

$f[i]=\sum\limits_{j=lst[i]}^{i-1}f[j]$

为什么？

观察序列：1，2，3，2，3，2

最后一颗珠子颜色为 $2$ ，它不能接在第 $1$ 到 $3$ 颗珠子后，因为这种情况（在第 $1$ 到 $3$ 颗珠子后接上颜色为 $2$ 的珠子）已经被第 $4$ 颗珠子统计过了。

$O(n\log n)$ 离散化， $O (n)$ 预处理出 $l s t$ 数组，前缀和优化 dp $O (n)$ ，总的时间复杂度 $O(n\log n)$ .

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 500005;
int n, m, c[N], lst[N];
LL f[N], b[N], a[N], ans = 0, sum[N];
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) scanf("%lld", &a[i]), b[i] = a[i];
    sort(b + 1, b + n + 1);
    for (int i = 1; i <= n; i++) a[i] = lower_bound(b + 1, b + n + 1, a[i]) - b;
    for (int i = 1; i <= n; i++) lst[i] = c[a[i]], c[a[i]] = i;
    for (int i = 1; i <= n; i++) {
        if (lst[i]) f[i] = (sum[i - 1] - sum[lst[i] - 1] + m) % m;
        else f[i] = (sum[i - 1] + 1) % m;
        sum[i] = (sum[i - 1] + f[i]) % m;
    }
    for (int i = 1; i <= n; i++) ans = (ans + f[i]) % m;
    printf("%lld", ans);
    return 0;
}