結果

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

ソースコード

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[2][N+10];
std::array<lli, N+10> A[2];

int main(void){
    int n; scanf("%d", &n);
    lli res = 0;
    int slen, plen = 0;
    REP(_, n){
        int working = (_ & 1), prev = 1 ^ working;
        scanf("%s", s[working]); slen = strlen(s[working]);
        int idx = 0; int l = (slen > plen) ? plen : slen;
        while(idx < l){
            if(s[working][idx] != s[prev][idx]) break;
            A[working][idx] = A[prev][idx] + 1;
            res += A[working][idx] * (A[working][idx] + 1) / 2;
            idx++;
        }
        while(idx < slen){
            A[working][idx] = 1;
            res++;
            idx++;
        }
        plen = slen;
    }
    printf("%lld\n", res);

    
    return 0;
}
0