1527C - Sequence Pair Weight

最新推荐文章于 2023-08-02 20:44:12 发布

Kyrie_10

最新推荐文章于 2023-08-02 20:44:12 发布

阅读量152

点赞数

分类专栏： dp

本文链接：https://blog.csdn.net/Kyrie_10/article/details/117304994

版权

dp 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

找出所有子串中 $a [i] = = a [j]$ 的数量

$d p [i]$ 表示以位置 $i$ 结尾的子串答案总和
$dp[i]=dp[i-1]+\sum{j(a[j]==a[i])}$
$j$ 代表可以选的子串左端点范围
数组为 $1 e 5$ ，答案爆int，使用long long
使用map复杂度为 $O (n l o g n)$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5;
ll dp[maxn];
int main()
{
    int t;
    scanf("%d", &t);
    while (t--)
    {
        int n, x;
        scanf("%d", &n);
        map <ll, ll> mp;
        ll ans = 0;
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &x);
            dp[i] = dp[i - 1] + mp[x];
            ans += dp[i];
            mp[x] += i;
        }
        printf("%lld\n", ans);
    }

    return 0;
}