1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
|
#include <bits/stdc++.h> #define FOR(i, a, b) for (int i = (a); i <= (b); ++i) #define ROF(i, a, b) for (int i = (a); i >= (b); --i) #define all(x) x.begin(),x.end() #define CLR(i,a) memset(i,a,sizeof(i)) #define fi first #define se second #define pb push_back #define SZ(a) ((int) a.size())
using namespace std;
typedef long long ll; typedef unsigned long long ull; typedef pair<ll,ll> pll; typedef array<ll,3> arr; typedef double DB; typedef pair<DB,DB> pdd; constexpr ll MAXN=static_cast<ll>(2e5)+10,INF=static_cast<ll>(5e18)+3;
ll N,M,T,A[MAXN],sumA[MAXN];
inline void solve(){ cin>>N>>M; FOR(i,1,M){ cin>>A[i]; } sort(A+1,A+1+M); sumA[0]=0; FOR(i,1,A[M]){ ll idx=M-(lower_bound(A+1,A+1+M,N-i)-A)+1; if(i==N){ idx=0; } sumA[i]=sumA[i-1]+idx; } ll ans=0; FOR(i,1,M){ ans+=sumA[A[i]]; if((A[i]-(N-A[i])+1)>0){ ans-=(A[i]-(N-A[i])+1); }
if(A[i]>=N){ ans+=2; } }
cout<<ans<<"\n"; }
int main() { ios::sync_with_stdio(false); cin.tie(0);cout.tie(0); cin>>T; while(T--){ solve(); #ifdef LOCAL cerr << '\n'; #endif } return 0; }
|