結果
問題 | No.962 LCPs |
ユーザー | uwi |
提出日時 | 2019-12-25 00:13:00 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 248 ms / 2,000 ms |
コード長 | 6,683 bytes |
コンパイル時間 | 4,677 ms |
コンパイル使用メモリ | 84,864 KB |
実行使用メモリ | 66,844 KB |
最終ジャッジ日時 | 2023-10-24 00:02:26 |
合計ジャッジ時間 | 16,180 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 52 ms
53,420 KB |
testcase_01 | AC | 53 ms
53,412 KB |
testcase_02 | AC | 56 ms
52,560 KB |
testcase_03 | AC | 54 ms
53,432 KB |
testcase_04 | AC | 53 ms
53,428 KB |
testcase_05 | AC | 53 ms
53,432 KB |
testcase_06 | AC | 53 ms
53,432 KB |
testcase_07 | AC | 55 ms
53,424 KB |
testcase_08 | AC | 55 ms
52,540 KB |
testcase_09 | AC | 55 ms
53,432 KB |
testcase_10 | AC | 56 ms
53,424 KB |
testcase_11 | AC | 55 ms
53,424 KB |
testcase_12 | AC | 54 ms
53,432 KB |
testcase_13 | AC | 53 ms
53,416 KB |
testcase_14 | AC | 55 ms
53,428 KB |
testcase_15 | AC | 54 ms
53,412 KB |
testcase_16 | AC | 54 ms
53,428 KB |
testcase_17 | AC | 248 ms
66,844 KB |
testcase_18 | AC | 189 ms
61,140 KB |
testcase_19 | AC | 194 ms
62,944 KB |
testcase_20 | AC | 169 ms
58,836 KB |
testcase_21 | AC | 166 ms
58,796 KB |
testcase_22 | AC | 159 ms
60,848 KB |
testcase_23 | AC | 158 ms
60,928 KB |
testcase_24 | AC | 150 ms
58,576 KB |
testcase_25 | AC | 153 ms
56,784 KB |
testcase_26 | AC | 147 ms
58,572 KB |
testcase_27 | AC | 163 ms
59,008 KB |
testcase_28 | AC | 147 ms
58,292 KB |
testcase_29 | AC | 141 ms
58,504 KB |
testcase_30 | AC | 160 ms
60,896 KB |
testcase_31 | AC | 174 ms
58,848 KB |
testcase_32 | AC | 139 ms
58,436 KB |
testcase_33 | AC | 194 ms
61,232 KB |
testcase_34 | AC | 153 ms
58,540 KB |
testcase_35 | AC | 135 ms
58,256 KB |
testcase_36 | AC | 157 ms
58,300 KB |
testcase_37 | AC | 166 ms
59,288 KB |
testcase_38 | AC | 165 ms
59,104 KB |
testcase_39 | AC | 161 ms
59,140 KB |
testcase_40 | AC | 161 ms
59,068 KB |
testcase_41 | AC | 164 ms
59,136 KB |
testcase_42 | AC | 172 ms
59,132 KB |
testcase_43 | AC | 165 ms
59,080 KB |
testcase_44 | AC | 167 ms
59,152 KB |
testcase_45 | AC | 165 ms
59,188 KB |
testcase_46 | AC | 170 ms
59,096 KB |
testcase_47 | AC | 99 ms
54,824 KB |
testcase_48 | AC | 96 ms
54,432 KB |
testcase_49 | AC | 101 ms
54,940 KB |
testcase_50 | AC | 101 ms
54,820 KB |
testcase_51 | AC | 98 ms
55,076 KB |
testcase_52 | AC | 92 ms
54,556 KB |
testcase_53 | AC | 99 ms
54,744 KB |
testcase_54 | AC | 95 ms
54,816 KB |
testcase_55 | AC | 82 ms
54,428 KB |
testcase_56 | AC | 101 ms
54,820 KB |
testcase_57 | AC | 110 ms
55,252 KB |
testcase_58 | AC | 108 ms
55,336 KB |
testcase_59 | AC | 104 ms
55,200 KB |
testcase_60 | AC | 110 ms
55,196 KB |
testcase_61 | AC | 110 ms
55,128 KB |
testcase_62 | AC | 111 ms
55,188 KB |
testcase_63 | AC | 217 ms
65,108 KB |
testcase_64 | AC | 100 ms
55,092 KB |
testcase_65 | AC | 224 ms
65,156 KB |
ソースコード
package adv2019; import java.io.ByteArrayInputStream; import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.ArrayList; import java.util.Arrays; import java.util.Comparator; import java.util.InputMismatchException; import java.util.List; public class D25 { InputStream is; PrintWriter out; String INPUT = ""; void solve() { int n = ni(); char[][] s = new char[n][]; for(int i = 0;i < n;i++) { s[i] = ns().toCharArray(); } int[] lcp = new int[n-1]; for(int i = 0;i < n-1;i++) { lcp[i] = lcp(s[i], s[i+1]); } int[][] li = new int[n-1][]; for(int i = 0;i < n-1;i++) { li[i] = new int[] {lcp[i], i+1}; } Arrays.sort(li, new Comparator<int[]>() { public int compare(int[] a, int[] b) { return a[0] - b[0]; } }); LST lst = new LST(n+1); lst.set(0).set(n); long ans = 0; for(int[] u : li) { int pos = u[1]; int pre = lst.prev(u[1]); int nex = lst.next(u[1]); long lnum = pos - pre; long rnum = nex - pos; long lsum = lnum*(lnum-1)/2; long rsum = rnum*(rnum-1)/2; ans += (lnum * rsum + rnum * lsum + lnum * rnum * 2) * u[0]; lst.set(pos); } for(char[] x : s) { ans += x.length; } out.println(ans); } public static class LST { public long[][] set; public int n; // public int size; public LST(int n) { this.n = n; int d = 1; for(int m = n;m > 1;m>>>=6, d++); set = new long[d][]; for(int i = 0, m = n>>>6;i < d;i++, m>>>=6){ set[i] = new long[m+1]; } // size = 0; } // [0,r) public LST setRange(int r) { for(int i = 0;i < set.length;i++, r=r+63>>>6){ for(int j = 0;j < r>>>6;j++){ set[i][j] = -1L; } if((r&63) != 0)set[i][r>>>6] |= (1L<<r)-1; } return this; } // [0,r) public LST unsetRange(int r) { if(r >= 0){ for(int i = 0;i < set.length;i++, r=r+63>>>6){ for(int j = 0;j < r+63>>>6;j++){ set[i][j] = 0; } if((r&63) != 0)set[i][r>>>6] &= ~((1L<<r)-1); } } return this; } public LST set(int pos) { if(pos >= 0 && pos < n){ // if(!get(pos))size++; for(int i = 0;i < set.length;i++, pos>>>=6){ set[i][pos>>>6] |= 1L<<pos; } } return this; } public LST unset(int pos) { if(pos >= 0 && pos < n){ // if(get(pos))size--; for(int i = 0;i < set.length && (i == 0 || set[i-1][pos] == 0L);i++, pos>>>=6){ set[i][pos>>>6] &= ~(1L<<pos); } } return this; } public boolean get(int pos) { return pos >= 0 && pos < n && set[0][pos>>>6]<<~pos<0; } public LST toggle(int pos) { return get(pos) ? unset(pos) : set(pos); } public int prev(int pos) { for(int i = 0;i < set.length && pos >= 0;i++, pos>>>=6, pos--){ int pre = prev(set[i][pos>>>6], pos&63); if(pre != -1){ pos = pos>>>6<<6|pre; while(i > 0)pos = pos<<6|63-Long.numberOfLeadingZeros(set[--i][pos]); return pos; } } return -1; } public int next(int pos) { for(int i = 0;i < set.length && pos>>>6 < set[i].length;i++, pos>>>=6, pos++){ int nex = next(set[i][pos>>>6], pos&63); if(nex != -1){ pos = pos>>>6<<6|nex; while(i > 0)pos = pos<<6|Long.numberOfTrailingZeros(set[--i][pos]); return pos; } } return -1; } private static int prev(long set, int n) { long h = set<<~n; if(h == 0L)return -1; return -Long.numberOfLeadingZeros(h)+n; } private static int next(long set, int n) { long h = set>>>n; if(h == 0L)return -1; return Long.numberOfTrailingZeros(h)+n; } @Override public String toString() { List<Integer> list = new ArrayList<Integer>(); for(int pos = next(0);pos != -1;pos = next(pos+1)){ list.add(pos); } return list.toString(); } } int lcp(char[] a, char[] b) { for(int i = 0;i < a.length && i < b.length;i++) { if(a[i] != b[i]) { return i; } } return Math.min(a.length, b.length); } void run() throws Exception { is = INPUT.isEmpty() ? System.in : new ByteArrayInputStream(INPUT.getBytes()); out = new PrintWriter(System.out); long s = System.currentTimeMillis(); solve(); out.flush(); if(!INPUT.isEmpty())tr(System.currentTimeMillis()-s+"ms"); // Thread t = new Thread(null, null, "~", Runtime.getRuntime().maxMemory()){ // @Override // public void run() { // long s = System.currentTimeMillis(); // solve(); // out.flush(); // if(!INPUT.isEmpty())tr(System.currentTimeMillis()-s+"ms"); // } // }; // t.start(); // t.join(); } public static void main(String[] args) throws Exception { new D25().run(); } private byte[] inbuf = new byte[1024]; public int lenbuf = 0, ptrbuf = 0; private int readByte() { if(lenbuf == -1)throw new InputMismatchException(); if(ptrbuf >= lenbuf){ ptrbuf = 0; try { lenbuf = is.read(inbuf); } catch (IOException e) { throw new InputMismatchException(); } if(lenbuf <= 0)return -1; } return inbuf[ptrbuf++]; } private boolean isSpaceChar(int c) { return !(c >= 33 && c <= 126); } private int skip() { int b; while((b = readByte()) != -1 && isSpaceChar(b)); return b; } private double nd() { return Double.parseDouble(ns()); } private char nc() { return (char)skip(); } private String ns() { int b = skip(); StringBuilder sb = new StringBuilder(); while(!(isSpaceChar(b))){ // when nextLine, (isSpaceChar(b) && b != ' ') sb.appendCodePoint(b); b = readByte(); } return sb.toString(); } private char[] ns(int n) { char[] buf = new char[n]; int b = skip(), p = 0; while(p < n && !(isSpaceChar(b))){ buf[p++] = (char)b; b = readByte(); } return n == p ? buf : Arrays.copyOf(buf, p); } private int[] na(int n) { int[] a = new int[n]; for(int i = 0;i < n;i++)a[i] = ni(); return a; } private long[] nal(int n) { long[] a = new long[n]; for(int i = 0;i < n;i++)a[i] = nl(); return a; } private char[][] nm(int n, int m) { char[][] map = new char[n][]; for(int i = 0;i < n;i++)map[i] = ns(m); return map; } private int[][] nmi(int n, int m) { int[][] map = new int[n][]; for(int i = 0;i < n;i++)map[i] = na(m); return map; } private int ni() { return (int)nl(); } private long nl() { long num = 0; int b; boolean minus = false; while((b = readByte()) != -1 && !((b >= '0' && b <= '9') || b == '-')); if(b == '-'){ minus = true; b = readByte(); } while(true){ if(b >= '0' && b <= '9'){ num = num * 10 + (b - '0'); }else{ return minus ? -num : num; } b = readByte(); } } private static void tr(Object... o) { System.out.println(Arrays.deepToString(o)); } }