题目描述

给定一个长度为𝑛n的序列 𝑎1,𝑎2,...,𝑎𝑛a1​,a2​,...,an​ ，请问多少种方案，能够从中选取一个连续段，使得该子段内所有元素的值都相同？

输入格式

输入共两行：
输入第一行，一个正整数𝑛n
输入第二行，𝑛n个整数𝑎1,𝑎2,...,𝑎𝑛a1​,a2​,...,an​ 。

输出格式

输出一个整数，表示方案数。

数据范围

• 对于 30%30% 的数据，1≤𝑛≤101≤n≤10
• 对于 60%60% 的数据，1≤𝑛≤1031≤n≤103
• 对于 100%100% 的数据，1≤𝑛≤1051≤n≤105 , −109≤𝑎𝑖≤109−109≤ai​≤109

样例数据

输入:

5
3 1 1 2 5

输出:

6

说明:

每个元素单独成段均可，共{3},{1},{1},{2},{5}这5种方案
第2、3个元素组成的子段{1,1}也满足要求。
所以共6种方案

详见代码:

#include <bits/stdc++.h>
using namespace std;
int a[100005];
long long ans = 0;
int n;
int main() 
{
    scanf("%d", &n);
    long long k;
    for (int i = 1; i <= n; i++) 
    {
        scanf("%d", &a[i]);
        if (i == 1)
        {
            k = 1;
        } 
        else if (a[i] == a[i - 1]) 
        {
            k++;
        } 
       else 
       {
            ans += (k * (k - 1) / 2);
            k = 1;
        }
    }
    ans += (k * (k - 1) / 2);
    cout << ans + n;
    return 0;
}