結果

問題 No.962 LCPs
コンテスト
ユーザー WarToks
提出日時 2020-01-01 21:49:00
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,265 bytes
コンパイル時間 1,034 ms
コンパイル使用メモリ 141,816 KB
実行使用メモリ 7,040 KB
最終ジャッジ日時 2024-11-30 15:59:15
合計ジャッジ時間 23,280 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 63 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <array>
#include <bitset>
#include <cstring>
#include <deque>
#include <math.h>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <queue>
#include <vector>

#define FOR(i, a, b) for(int (i) = (a); (i) < (b); ++(i))
#define rFOR(i, a, b) for(int (i) = (b); (i) >= (a); --(i))
#define REP(i, n) FOR(i, 0, n)
#define rREP(i, n) rFOR(i, 0, (n-1))
#define SORT(A) std::sort((A).begin(), (A).end())
#define ALL(A) (A).begin(), (A).end()
// 座標圧縮 (for vector) : ソートしてから使うのが一般的 ; SORT(A) => COORDINATE_COMPRESSION(A)
#define COORDINATE_COMPRESSION(A) (A).erase(unique((A).begin(),(A).end()),(A).end())

using lli = long long int;
using pii = std::pair<int, int>;

void VintOut(std::vector<int>& A){
    const int n = A.size();
    if(n == 0){putchar('\n'); return;}
    printf("%d", A[0]);
    for(int i = 1; i < n; ++i) printf(" %d", A[i]);
    putchar('\n');
}
void VintOut(std::vector<long long int>& A){
    const int n = A.size();
    if(n == 0){ putchar('\n'); return;}
    printf("%lld", A[0]);
    for(int i = 1; i < n; ++i) printf(" %lld", A[i]);
    putchar('\n');
}

template <typename T>
inline bool chmin(T& a, T b){
    if(b < a){ a = b; return true;}
    return false;
}

template <typename T>
inline bool chmax(T& a, T b){
    if(a < b){ a = b; return true;}
    return false;
}


// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

constexpr int N = 2e5;
char s[N+10], ps[N+10];
std::vector<lli> Prev, Working;

int main(void){
    int n; scanf("%d", &n);
    lli res = 0;
    int slen, plen = 0;
    Prev.resize(N+10); Working.resize(N+10);
    REP(_, n){
        scanf("%s", s); slen = strlen(s);
        //Working.resize(slen);
        int idx = 0; int l = (slen > plen) ? plen : slen;
        while(idx < l){
            if(s[idx] != ps[idx]) break;
            Working[idx] = Prev[idx] + 1;
            res += Working[idx] * (Working[idx] + 1) / 2;
            idx++;
        }
        while(idx < slen){
            Working[idx] = 1;
            res++;
            idx++;
        }
        std::swap(Working, Prev);
        std::swap(s, ps); plen = slen;
    }
    printf("%lld\n", res);

    
    return 0;
}
0