#pragma GCC optimize ("Ofast") #include using namespace std; void *wmem; char memarr[96000000]; template inline void walloc1d(T **arr, int x, void **mem = &wmem){ static int skip[16] = {0, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1}; (*mem) = (void*)( ((char*)(*mem)) + skip[((unsigned long long)(*mem)) & 15] ); (*arr)=(T*)(*mem); (*mem)=((*arr)+x); } template void sortA_L(int N, T1 a[], void *mem = wmem){ sort(a, a+N); } template void sortA_L(int N, T1 a[], T2 b[], void *mem = wmem){ int i; pair *arr; walloc1d(&arr, N, &mem); for(i=(0);i<(N);i++){ arr[i].first = a[i]; arr[i].second = b[i]; } sort(arr, arr+N); for(i=(0);i<(N);i++){ a[i] = arr[i].first; b[i] = arr[i].second; } } inline int my_getchar_unlocked(){ static char buf[1048576]; static int s = 1048576; static int e = 1048576; if(s == e && e == 1048576){ e = fread_unlocked(buf, 1, 1048576, stdin); s = 0; } if(s == e){ return EOF; } return buf[s++]; } inline void rd(int &x){ int k; int m=0; x=0; for(;;){ k = my_getchar_unlocked(); if(k=='-'){ m=1; break; } if('0'<=k&&k<='9'){ x=k-'0'; break; } } for(;;){ k = my_getchar_unlocked(); if(k<'0'||k>'9'){ break; } x=x*10+k-'0'; } if(m){ x=-x; } } struct MY_WRITER{ char buf[1048576]; int s; int e; MY_WRITER(){ s = 0; e = 1048576; } ~MY_WRITER(){ if(s){ fwrite_unlocked(buf, 1, s, stdout); } } } ; MY_WRITER MY_WRITER_VAR; void my_putchar_unlocked(int a){ if(MY_WRITER_VAR.s == MY_WRITER_VAR.e){ fwrite_unlocked(MY_WRITER_VAR.buf, 1, MY_WRITER_VAR.s, stdout); MY_WRITER_VAR.s = 0; } MY_WRITER_VAR.buf[MY_WRITER_VAR.s++] = a; } inline void wt_L(char a){ my_putchar_unlocked(a); } inline void wt_L(long long x){ int s=0; int m=0; char f[20]; if(x<0){ m=1; x=-x; } while(x){ f[s++]=x%10; x/=10; } if(!s){ f[s++]=0; } if(m){ my_putchar_unlocked('-'); } while(s--){ my_putchar_unlocked(f[s]+'0'); } } int N; int K; int A[100000]; int val[100000]; int p[100000]; int vs; map mp; vector vi[100000]; vector vsm[100000]; void calc(int a, int l, int r, int &cnt, long long &sm){ int ind; int s; int t; cnt = 0; sm = 0; if(mp.count(a)==0){ return; } ind = mp[a]; s = lower_bound(vi[ind].begin(), vi[ind].end(), l) - vi[ind].begin(); t = upper_bound(vi[ind].begin(), vi[ind].end(), r) - vi[ind].begin(); cnt = t - s; sm = vsm[ind][t] - vsm[ind][s]; } int main(){ int m; wmem = memarr; int i; int j; int k; int v; int x; int y; int z; int c; long long res = 0; long long tmp; set s; set::iterator it1; set::iterator it2; rd(N); rd(K); { int Lj4PdHRW; for(Lj4PdHRW=(0);Lj4PdHRW<(N);Lj4PdHRW++){ rd(A[Lj4PdHRW]); } } for(i=(0);i<(N);i++){ val[i] = A[i]; p[i] = i; } sortA_L(N, val, p); k = 0; for(i=(0);i<(N);i++){ if(mp.count(A[i])==0){ mp[A[i]] = k++; } } for(i=(0);i<(N);i++){ vi[mp[A[i]]].push_back(i); } for(i=(0);i<(k);i++){ tmp = 0; vsm[i].push_back(tmp); for(j=(0);j<(vi[i].size());j++){ vsm[i].push_back( tmp += vi[i][j] ); } } s.insert(-1); s.insert(N); for(m=(0);m<(N);m++){ y = k = p[m]; v = val[m]; while(vs < N && val[vs] <= v){ s.insert(p[vs]); vs++; } it1 = it2 = s.lower_bound(k); it1--; it2++; x = (*it1) + 1; z = (*it2) - 1; if(y-x > z-y){ for(i=(y);i<(z+1);i++){ calc(K - A[i], x, y, c, tmp); res += (long long)(i+1) * c - tmp; } } else{ for(i=(x);i<(y+1);i++){ calc(K - A[i], y, z, c, tmp); res += tmp - (long long)(i-1) * c; } } } wt_L(res); wt_L('\n'); return 0; } // cLay varsion 20200509-1 // --- original code --- // int N, K, A[1d5]; // int val[1d5], p[1d5], vs; // map mp; // vector vi[1d5]; // vector vsm[1d5]; // // void calc(int a, int l, int r, int &cnt, ll &sm){ // int ind, s, t; // cnt = 0; // sm = 0; // if(mp.count(a)==0) return; // ind = mp[a]; // // s = lower_bound(vi[ind].begin(), vi[ind].end(), l) - vi[ind].begin(); // t = upper_bound(vi[ind].begin(), vi[ind].end(), r) - vi[ind].begin(); // cnt = t - s; // sm = vsm[ind][t] - vsm[ind][s]; // } // // { // int i, j, k, v, x, y, z, c; // ll res = 0, tmp; // set s; // set::iterator it1, it2; // rd(N,K,A(N)); // // rep(i,N) val[i] = A[i], p[i] = i; // sortA(N, val, p); // // k = 0; // rep(i,N) if(mp.count(A[i])==0) mp[A[i]] = k++; // rep(i,N) vi[mp[A[i]]].push_back(i); // rep(i,k){ // tmp = 0; // vsm[i].push_back(tmp); // rep(j,vi[i].size()) vsm[i].push_back( tmp += vi[i][j] ); // } // // s.insert(-1); // s.insert(N); // rep(m,N){ // y = k = p[m]; // v = val[m]; // while(vs < N && val[vs] <= v){ // s.insert(p[vs]); // vs++; // } // it1 = it2 = s.lower_bound(k); // it1--; // it2++; // x = (*it1) + 1; // z = (*it2) - 1; // // // wt(v,":",x,y,z); // if(y-x > z-y){ // rep(i,y,z+1){ // calc(K - A[i], x, y, c, tmp); // res += (ll)(i+1) * c - tmp; // } // } else { // rep(i,x,y+1){ // calc(K - A[i], y, z, c, tmp); // res += tmp - (ll)(i-1) * c; // } // } // } // // wt(res); // }