题目描述
给定一个长度为𝑛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;
}
)
