fork download
copy 
  1. import math
  2. def ip():
  3.     return map(int,input().split())
  4.  
  5. T=int(input())
  6. # T=1
  7. for __ in range(T):
  8.     n,m=ip()
  9.     a=list(ip())
  10.  
  11.  
  12.     lst=[0 for _ in range(n+1)]
  13.     dic={}
  14.     for i in a:
  15.         if i not in dic:
  16.             dic[i]=0
  17.         dic[i]+=1
  18.     # print(dic)
  19.     for i in range(n,0,-1):
  20.         if i==n:
  21.             lst[i]=dic.get(i,0)
  22.             continue
  23.         lst[i]=lst[i+1]+dic.get(i,0)
  24.         # print(lst)
  25.  
  26.  
  27.  
  28.  
  29.     ct=0
  30.     for i in range(1,n):
  31.         x=lst[i]
  32.         y=lst[n-i]
  33.  
  34.         ct+=(x*y)-min(x,y)
  35.  
  36.     print(ct)
  37.  
  38.  
  39.  
  40.  
  41.  
  42.  
  43.  
  44.  
  45.  
Success #stdin #stdout 0.09s 14152KB
comments (?)
stdin
copy 
3
5 2
2 4
5 2
3 4
12 3
5 9 8
stdout
copy 
4
6
22
https://ideone.com/JXEzr7
language:
Python 3 (python  3.12)
created:
12 hours ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!