結果

問題 No.962 LCPs
コンテスト
ユーザー Leonardone
提出日時 2019-12-25 19:45:57
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,619 bytes
コンパイル時間 707 ms
コンパイル使用メモリ 73,432 KB
実行使用メモリ 9,984 KB
最終ジャッジ日時 2024-09-25 10:01:39
合計ジャッジ時間 3,382 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 51 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

void bins(int col, int minRow, int maxRow);

int n;
vector<string> vs(2e5);
long long ans;

int main() {
    cin >> n;
    vs.resize(n);
    for (int i = 0; i < n; i++) {
        cin >> vs[i];
    }
    // sort(vs.begin(), vs.end());
    
    // for (int i = 0; i < n; i++) cout << vs[i] << endl;
    
    bins(0, 0, n-1);
    
    cout << ans << endl;
    return 0;
}

void bins(int col, int minRow, int maxRow) {
    int lRow = minRow;
    for (int row = minRow; row <= maxRow; row++) {
        if (vs[lRow].size() <= col) {
            lRow = row; continue;
        }
        if (vs[row].size() <= col) {
            lRow = row; continue;
        } else if (vs[row][col] == vs[lRow][col]) {
            continue;
        }
        int len = row - lRow;
        if (len == 1) {
            ans += (long long)vs[lRow].size() - (long long)col;
        } else {
            long long lp = 1, rp = len;
            while (lp <= rp) {
                ans += lp * rp * (lp == rp ? 1LL : 2LL);
                lp++; rp--;
            }
            bins(col + 1, lRow, row - 1);
        }
        lRow = row;
    }
    if (vs[lRow].size() >= col) {
        int row = maxRow + 1;
        int len = row - lRow;
        if (len == 1) {
            ans += (long long)vs[lRow].size() - (long long)col;
        } else {
            long long lp = 1, rp = len;
            while (lp <= rp) {
                ans += lp * rp * (lp == rp ? 1LL : 2LL);
                lp++; rp--;
            }
            bins(col + 1, lRow, row - 1);
        }
    }
}
0