結果
問題 | No.1347 HS Railway |
ユーザー | LayCurse |
提出日時 | 2021-01-16 13:45:49 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 59 ms / 2,000 ms |
コード長 | 17,866 bytes |
コンパイル時間 | 2,951 ms |
コンパイル使用メモリ | 220,476 KB |
実行使用メモリ | 8,448 KB |
最終ジャッジ日時 | 2024-11-27 14:28:11 |
合計ジャッジ時間 | 5,278 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 1 ms
5,248 KB |
testcase_08 | AC | 2 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 2 ms
5,248 KB |
testcase_12 | AC | 9 ms
5,504 KB |
testcase_13 | AC | 9 ms
5,248 KB |
testcase_14 | AC | 5 ms
5,248 KB |
testcase_15 | AC | 4 ms
5,248 KB |
testcase_16 | AC | 13 ms
6,144 KB |
testcase_17 | AC | 4 ms
5,248 KB |
testcase_18 | AC | 20 ms
6,912 KB |
testcase_19 | AC | 22 ms
6,400 KB |
testcase_20 | AC | 13 ms
6,528 KB |
testcase_21 | AC | 4 ms
5,248 KB |
testcase_22 | AC | 18 ms
7,016 KB |
testcase_23 | AC | 20 ms
6,764 KB |
testcase_24 | AC | 7 ms
5,248 KB |
testcase_25 | AC | 13 ms
5,888 KB |
testcase_26 | AC | 19 ms
6,912 KB |
testcase_27 | AC | 13 ms
5,632 KB |
testcase_28 | AC | 15 ms
7,168 KB |
testcase_29 | AC | 19 ms
6,888 KB |
testcase_30 | AC | 15 ms
6,784 KB |
testcase_31 | AC | 15 ms
6,492 KB |
testcase_32 | AC | 25 ms
7,276 KB |
testcase_33 | AC | 25 ms
7,336 KB |
testcase_34 | AC | 28 ms
7,168 KB |
testcase_35 | AC | 22 ms
7,296 KB |
testcase_36 | AC | 26 ms
7,184 KB |
testcase_37 | AC | 27 ms
7,592 KB |
testcase_38 | AC | 30 ms
7,640 KB |
testcase_39 | AC | 32 ms
7,552 KB |
testcase_40 | AC | 21 ms
7,168 KB |
testcase_41 | AC | 23 ms
6,916 KB |
testcase_42 | AC | 59 ms
7,776 KB |
testcase_43 | AC | 31 ms
7,296 KB |
testcase_44 | AC | 41 ms
7,324 KB |
testcase_45 | AC | 37 ms
7,064 KB |
testcase_46 | AC | 39 ms
7,552 KB |
testcase_47 | AC | 57 ms
7,680 KB |
testcase_48 | AC | 58 ms
8,448 KB |
testcase_49 | AC | 45 ms
7,676 KB |
testcase_50 | AC | 54 ms
7,936 KB |
testcase_51 | AC | 47 ms
7,552 KB |
ソースコード
#pragma GCC optimize ("Ofast") #include<bits/stdc++.h> using namespace std; void*wmem; char memarr[96000000]; template<class S, class T> inline S min_L(S a,T b){ return a<=b?a:b; } template<class S, class T> inline S max_L(S a,T b){ return a>=b?a:b; } template<class T> 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<class T> inline void walloc1d(T **arr, int x1, int x2, void **mem = &wmem){ walloc1d(arr, x2-x1, mem); (*arr) -= x1; } 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; } } inline void rd(long long &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'); } } template<class T> struct Arr1d{ int n; int mem; T*d; T& operator[](int a){ return d[a]; } void sort(){ reset(); std::sort(d, d+n); } int set_cumulative_sum; int cumulative_sum_mem; T*cumulative_sum; void setSum(void){ int i; set_cumulative_sum = 1; if(cumulative_sum_mem < n+1){ delete[] cumulative_sum; cumulative_sum = new T[n+1]; cumulative_sum_mem = n+1; } cumulative_sum[0] = 0; for(i=(0);i<(n);i++){ cumulative_sum[i+1] = cumulative_sum[i] + d[i]; } } T getSum(int i, int j){ if(set_cumulative_sum==0){ setSum(); } return cumulative_sum[j+1] - cumulative_sum[i]; } int set_const_len_left; int const_len_left_mem; int*const_len_left; void setConstLenLeft(void){ int i; set_const_len_left = 1; if(const_len_left_mem < n){ delete[] const_len_left; const_len_left = new int[n]; const_len_left_mem = n; } for(i=(0);i<(n);i++){ const_len_left[i] = 1; } for(i=(1);i<(n);i++){ if(d[i]==d[i-1]){ const_len_left[i] = const_len_left[i-1] + 1; } } } int ConstLenLeft(int st, T val){ if(!set_const_len_left){ setConstLenLeft(); } if(val != d[st]){ return 0; } return const_len_left[st]; } int ConstLenLeft(int st){ if(!set_const_len_left){ setConstLenLeft(); } return const_len_left[st]; } int ConstLenLeftCyclic(int st, T val){ if(!set_const_len_left){ setConstLenLeft(); } st %= n; if(st < 0){ st += n; } if(val != d[st]){ return 0; } if(const_len_left[st] != st+1 || d[st] != d[n-1]){ return const_len_left[st]; } if(const_len_left[n-1] == n){ return 1073709056; } return const_len_left[st] + const_len_left[n-1]; } int ConstLenLeftCyclic(int st){ if(!set_const_len_left){ setConstLenLeft(); } st %= n; if(st < 0){ st += n; } if(const_len_left[st] != st+1 || d[st] != d[n-1]){ return const_len_left[st]; } if(const_len_left[n-1] == n){ return 1073709056; } return const_len_left[st] + const_len_left[n-1]; } int set_const_len_right; int const_len_right_mem; int*const_len_right; void setConstLenRight(void){ int i; set_const_len_right = 1; if(const_len_right_mem < n){ delete[] const_len_right; const_len_right = new int[n]; const_len_right_mem = n; } for(i=(0);i<(n);i++){ const_len_right[i] = 1; } for(i=(n-1)-1;i>=(0);i--){ if(d[i]==d[i+1]){ const_len_right[i] = const_len_right[i+1] + 1; } } } int ConstLenRight(int st, T val){ if(!set_const_len_right){ setConstLenRight(); } if(val != d[st]){ return 0; } return const_len_right[st]; } int ConstLenRight(int st){ if(!set_const_len_right){ setConstLenRight(); } return const_len_right[st]; } int ConstLenRightCyclic(int st, T val){ if(!set_const_len_right){ setConstLenRight(); } if(val != d[st]){ return 0; } st %= n; if(st < 0){ st += n; } if(const_len_right[st] != n-st || d[st] != d[0]){ return const_len_right[st]; } if(const_len_right[0] == n){ return 1073709056; } return const_len_right[st] + const_len_right[0]; } int ConstLenRightCyclic(int st){ if(!set_const_len_right){ setConstLenRight(); } st %= n; if(st < 0){ st += n; } if(const_len_right[st] != n-st || d[st] != d[0]){ return const_len_right[st]; } if(const_len_right[0] == n){ return 1073709056; } return const_len_right[st] + const_len_right[0]; } int set_dhist; int dhist_mem; int*dhist; int*dhists; T dhist_mn; T dhist_mx; void setDHist(void){ int i; int len; set_dhist = 1; if(n==0){ return; } dhist_mn = dhist_mx = d[0]; for(i=(1);i<(n);i++){ if(dhist_mn > d[i]){ dhist_mn = d[i]; } if(dhist_mx < d[i]){ dhist_mx = d[i]; } } len = dhist_mx - dhist_mn + 1; if(dhist_mem < len){ delete[] dhist; dhist = new int[len]; delete[] dhists; dhists = new int[len+1]; dhist_mem = len; } for(i=(0);i<(len);i++){ dhist[i] = 0; } for(i=(0);i<(n);i++){ dhist[d[i] - dhist_mn]++; } dhists[0] = 0; for(i=(0);i<(len);i++){ dhists[i+1] = dhists[i] + dhist[i]; } } int dHist(T x){ if(set_dhist==0){ setDHist(); } if(n == 0 || x < dhist_mn || x > dhist_mx){ return 0; } return dhist[x - dhist_mn]; } int dHist(T x, T y){ if(set_dhist==0){ setDHist(); } if(x < dhist_mn){ x = dhist_mn; } if(y > dhist_mx){ y = dhist_mx; } if(n == 0 || x > y){ return 0; } return dhists[y-dhist_mn+1] - dhists[x-dhist_mn]; } int set_shist; int shist_mem; T*shist; void setSHist(void){ int i; set_shist = 1; if(shist_mem < n){ delete[] shist; shist = new T[n]; shist_mem = n; } for(i=(0);i<(n);i++){ shist[i] = d[i]; } std::sort(shist, shist + n); } int sHist(T x){ if(set_shist==0){ setSHist(); } auto r = equal_range(shist, shist+n, x); return r.second - r.first; } int sHist(T x, T y){ if(set_shist==0){ setSHist(); } return upper_bound(shist, shist+n, y) - lower_bound(shist, shist+n, x); } int set_prevLE; int prevLE_mem; int*prevLE; int setPrevLE(void *mem = wmem){ int i; int s = 0; int*st; set_prevLE = 1; if(prevLE_mem < n){ delete[] prevLE; prevLE = new int[n]; prevLE_mem = n; } walloc1d(&st, n, &mem); for(i=(0);i<(n);i++){ while(s && d[st[s-1]] > d[i]){ s--; } if(s==0){ prevLE[i] = -1; } else{ prevLE[i] = st[s-1]; } st[s++] = i; } } int PrevLE(int i){ if(set_prevLE==0){ setPrevLE(); } return prevLE[i]; } int set_prevLT; int prevLT_mem; int*prevLT; int setPrevLT(void *mem = wmem){ int i; int s = 0; int*st; set_prevLT = 1; if(prevLT_mem < n){ delete[] prevLT; prevLT = new int[n]; prevLT_mem = n; } walloc1d(&st, n, &mem); for(i=(0);i<(n);i++){ while(s && d[st[s-1]] >= d[i]){ s--; } if(s==0){ prevLT[i] = -1; } else{ prevLT[i] = st[s-1]; } st[s++] = i; } } int PrevLT(int i){ if(set_prevLT==0){ setPrevLT(); } return prevLT[i]; } int set_prevGE; int prevGE_mem; int*prevGE; int setPrevGE(void *mem = wmem){ int i; int s = 0; int*st; set_prevGE = 1; if(prevGE_mem < n){ delete[] prevGE; prevGE = new int[n]; prevGE_mem = n; } walloc1d(&st, n, &mem); for(i=(0);i<(n);i++){ while(s && d[st[s-1]] < d[i]){ s--; } if(s==0){ prevGE[i] = -1; } else{ prevGE[i] = st[s-1]; } st[s++] = i; } } int PrevGE(int i){ if(set_prevGE==0){ setPrevGE(); } return prevGE[i]; } int set_prevGT; int prevGT_mem; int*prevGT; int setPrevGT(void *mem = wmem){ int i; int s = 0; int*st; set_prevGT = 1; if(prevGT_mem < n){ delete[] prevGT; prevGT = new int[n]; prevGT_mem = n; } walloc1d(&st, n, &mem); for(i=(0);i<(n);i++){ while(s && d[st[s-1]] <= d[i]){ s--; } if(s==0){ prevGT[i] = -1; } else{ prevGT[i] = st[s-1]; } st[s++] = i; } } int PrevGT(int i){ if(set_prevGT==0){ setPrevGT(); } return prevGT[i]; } int set_nextLE; int nextLE_mem; int*nextLE; int setNextLE(void *mem = wmem){ int i; int s = 0; int*st; set_nextLE = 1; if(nextLE_mem < n){ delete[] nextLE; nextLE = new int[n]; nextLE_mem = n; } walloc1d(&st, n, &mem); for(i=(n)-1;i>=(0);i--){ while(s && d[st[s-1]] > d[i]){ s--; } if(s==0){ nextLE[i] = n; } else{ nextLE[i] = st[s-1]; } st[s++] = i; } } int NextLE(int i){ if(set_nextLE==0){ setNextLE(); } return nextLE[i]; } int set_nextLT; int nextLT_mem; int*nextLT; int setNextLT(void *mem = wmem){ int i; int s = 0; int*st; set_nextLT = 1; if(nextLT_mem < n){ delete[] nextLT; nextLT = new int[n]; nextLT_mem = n; } walloc1d(&st, n, &mem); for(i=(n)-1;i>=(0);i--){ while(s && d[st[s-1]] >= d[i]){ s--; } if(s==0){ nextLT[i] = n; } else{ nextLT[i] = st[s-1]; } st[s++] = i; } } int NextLT(int i){ if(set_nextLT==0){ setNextLT(); } return nextLT[i]; } int set_nextGE; int nextGE_mem; int*nextGE; int setNextGE(void *mem = wmem){ int i; int s = 0; int*st; set_nextGE = 1; if(nextGE_mem < n){ delete[] nextGE; nextGE = new int[n]; nextGE_mem = n; } walloc1d(&st, n, &mem); for(i=(n)-1;i>=(0);i--){ while(s && d[st[s-1]] < d[i]){ s--; } if(s==0){ nextGE[i] = n; } else{ nextGE[i] = st[s-1]; } st[s++] = i; } } int NextGE(int i){ if(set_nextGE==0){ setNextGE(); } return nextGE[i]; } int set_nextGT; int nextGT_mem; int*nextGT; int setNextGT(void *mem = wmem){ int i; int s = 0; int*st; set_nextGT = 1; if(nextGT_mem < n){ delete[] nextGT; nextGT = new int[n]; nextGT_mem = n; } walloc1d(&st, n, &mem); for(i=(n)-1;i>=(0);i--){ while(s && d[st[s-1]] <= d[i]){ s--; } if(s==0){ nextGT[i] = n; } else{ nextGT[i] = st[s-1]; } st[s++] = i; } } int NextGT(int i){ if(set_nextGT==0){ setNextGT(); } return nextGT[i]; } void reset(){ set_cumulative_sum = 0; set_const_len_left = 0; set_const_len_right = 0; set_dhist = 0; set_shist = 0; set_prevLE = set_prevLT = set_prevGE = set_prevGT = 0; set_nextLE = set_nextLT = set_nextGE = set_nextGT = 0; } void memory_expand(int nn){ if(mem < nn){ delete[] d; d = new T[nn]; mem = nn; } } void malloc(int nn){ reset(); memory_expand(nn); n = nn; } void setN(int nn){ reset(); memory_expand(nn); n = nn; } void set(vector<T> &a){ int i; int nn = a.size(); setN(nn); for(i=(0);i<(nn);i++){ d[i] = a[i]; } } void set(int nn, T a[]){ int i; setN(nn); for(i=(0);i<(nn);i++){ d[i] = a[i]; } } void free(){ destructor(); } void constructor(){ n = mem = 0; d = NULL; set_cumulative_sum = 0; cumulative_sum_mem = 0; cumulative_sum = NULL; set_const_len_left = 0; const_len_left_mem = 0; const_len_left = NULL; set_const_len_right = 0; const_len_right_mem = 0; const_len_right = NULL; set_dhist = 0; dhist_mem = 0; dhist = NULL; dhists = NULL; set_shist = 0; shist_mem = 0; shist = NULL; set_prevLE = set_prevLT = set_prevGE = set_prevGT = 0; prevLE_mem = prevLT_mem = prevGE_mem = prevGT_mem = 0; prevLE = prevLT = prevGE = prevGT = NULL; set_nextLE = set_nextLT = set_nextGE = set_nextGT = 0; nextLE_mem = nextLT_mem = nextGE_mem = nextGT_mem = 0; nextLE = nextLT = nextGE = nextGT = NULL; } void constructor(int nn){ constructor(); malloc(nn); } void destructor(){ delete[] d; d = NULL; mem = n = 0; set_cumulative_sum = 0; cumulative_sum_mem = 0; delete[] cumulative_sum; cumulative_sum = NULL; set_const_len_left = 0; const_len_left_mem = 0; delete[] const_len_left; const_len_left = NULL; set_const_len_right = 0; const_len_right_mem = 0; delete[] const_len_right; const_len_right = NULL; set_dhist = 0; dhist_mem = 0; delete[] dhist; delete[] dhists; dhist = NULL; set_shist = 0; shist_mem = 0; delete[] shist; shist = NULL; set_prevLE = set_prevLT = set_prevGE = set_prevGT = 0; prevLE_mem = prevLT_mem = prevGE_mem = prevGT_mem = 0; delete[] prevLE; delete[] prevLT; delete[] prevGE; delete[] prevGT; prevLE = prevLT = prevGE = prevGT = NULL; set_nextLE = set_nextLT = set_nextGE = set_nextGT = 0; nextLE_mem = nextLT_mem = nextGE_mem = nextGT_mem = 0; delete[] nextLE; delete[] nextLT; delete[] nextGE; delete[] nextGT; nextLE = nextLT = nextGE = nextGT = NULL; } Arr1d(){ constructor(); } Arr1d(int nn){ constructor(nn); } ~Arr1d(){ destructor(); } } ; int N; int M; int L[5]; int R[5]; long long A[5][100000]; int B[100000]; long long S[100000]; int xs; long long xt[100000]; int ys; long long yt[100000]; Arr1d<long long> tmp; int main(){ wmem = memarr; int i; int j; int k; int x; int y; int l; int r; long long res = 0; long long z; rd(N); for(i=(0);i<(5);i++){ rd(L[i]);L[i] += (-1); rd(R[i]);R[i] += (-1); for(k=(L[i]);k<(R[i]);k++){ rd(A[i][k]); } } rd(M); { int cTE1_r3A; for(cTE1_r3A=(0);cTE1_r3A<(M);cTE1_r3A++){ rd(B[cTE1_r3A]);B[cTE1_r3A] += (-1); rd(S[cTE1_r3A]);S[cTE1_r3A] += (-1); } } for(x=(0);x<(5);x++){ for(y=(x+1);y<(5);y++){ xs = ys = 0; for(i=(0);i<(M);i++){ if(B[i]==x){ xt[xs++] = S[i]; } if(B[i]==y){ yt[ys++] = S[i]; } } l =max_L(L[x], L[y]); r =min_L(R[x], R[y]); if(l >= r){ continue; } z = 0; for(i=(L[x]);i<(l);i++){ z += A[x][i]; } for(i=(0);i<(xs);i++){ xt[i] += z; } z = 0; for(i=(L[y]);i<(l);i++){ z += A[y][i]; } for(i=(0);i<(ys);i++){ yt[i] += z; } z = 0; for(i=(l);i<(r);i++){ z += A[y][i] - A[x][i]; } tmp.set(ys, yt); for(i=(0);i<(xs);i++){ res += tmp.sHist(xt[i]-z, xt[i]); } } } wt_L(res); wt_L('\n'); return 0; } // cLay version 20210103-1 [bug fixed 1] // --- original code --- // int N, M, L[5], R[5]; ll A[5][1d5]; // int B[1d5]; ll S[1d5]; // int xs; ll xt[1d5]; // int ys; ll yt[1d5]; // Arr1d<ll> tmp; // { // int i, j, k, x, y, l, r; // ll res = 0, z; // rd(N); // rep(i,5){ // rd(L[i]--, R[i]--); // rep(k,L[i],R[i]) rd(A[i][k]); // } // rd(M,(B--,S--)(M)); // // rep(x,5) rep(y,x+1,5){ // xs = ys = 0; // rep(i,M){ // if(B[i]==x) xt[xs++] = S[i]; // if(B[i]==y) yt[ys++] = S[i]; // } // l = max(L[x], L[y]); // r = min(R[x], R[y]); // if(l >= r) continue; // // // wt(x, y, xs, ys); // // z = 0; // rep(i,L[x],l) z += A[x][i]; // rep(i,xs) xt[i] += z; // // z = 0; // rep(i,L[y],l) z += A[y][i]; // rep(i,ys) yt[i] += z; // // z = 0; // rep(i,l,r) z += A[y][i] - A[x][i]; // // // wt(xt(xs)); // // wt(yt(ys)); // // wt(z); // // tmp.set(ys, yt); // rep(i,xs) res += tmp.sHist(xt[i]-z, xt[i]); // } // wt(res); // }