MT1415 大小相同

题目

给定一个由N(<10)个正整数组成的数组A，生成一些最小元素和最大元素相同的子数组数（可以仅包含1个元素），统计这些子数组的数量并输出。

注：最大元素和最小元素相同就是数组中的元素全部为同一个值。如数组（1,1），它满足条件的子数组有三个，分别为（1）（1）（1,1）

格式

输入格式

第一行输入数组长度N，第二行输入数组元素，整型，空格分隔。

输出格式

输出整型

样例1

输入：
3
1 1 3输出：
4

程序

def main():#code heren = int(input())li  = list(map(int,input().split()))# 统计每个元素的出现次数count = {}for i in li:if i in count:count[i] += 1else:count[i] = 1res = 0for i in count:res += (2**count[i])-1print(res)if __name__ == '__main__':main();