結果

問題 No.962 LCPs
ユーザー Leonardone
提出日時 2019-12-25 09:15:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,388 bytes
コンパイル時間 1,006 ms
コンパイル使用メモリ 82,372 KB
実行使用メモリ 9,992 KB
最終ジャッジ日時 2024-09-24 13:47:35
合計ジャッジ時間 3,686 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 23 WA * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using namespace std;

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

int n;
vector<string> vs(2e5);
unsigned 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 hRow = 0;
        for (int b=25; b >=0; b--) {
            int v = hRow | (1 << b);
            if (minRow <= v && v <= maxRow) {
                if (vs[v].size() <= col) {
                    hRow = v;
                }
            }
        }
        if (minRow <= hRow && hRow < maxRow && vs[hRow].size() <= col) {
            unsigned long long len = hRow - minRow + 1;
            // cout << "[" << col << "] " << "nosize: " <<  hRow << ", (" << len << ")" << endl;
            unsigned long long lp = 1, rp = len;
            while (lp <= rp) {
                ans += lp * rp * (lp == rp ? 1ULL : 2ULL);
                lp++; rp--;
            }
        }
    }
    for (char ch = 'a'; ch <= 'z'; ch++) {
        int lRow = 0, hRow = 0;
        for (int b=25; b >=0; b--) {
            int u = lRow | (1 << b);
            int v = hRow | (1 << b);
            if (minRow <= u && u <= maxRow) {
                if (vs[u].size() <= col || vs[u][col] < ch) {
                    lRow = u;
                }
            }
            if (minRow <= v && v <= maxRow) {
                if (vs[v].size() <= col || vs[v][col] <= ch) {
                    hRow = v;
                }
            }
        }
        if (vs[hRow].size() <= col || vs[hRow][col] != ch) { continue; }
        if (vs[lRow].size() <= col || vs[lRow][col] != ch) { lRow++; }
        unsigned long long len = hRow - lRow + 1;
        // cout << "[" << col << "] " << ch << ": " << lRow << ", " << hRow << ", (" << len << ")" << endl;
        if (len == 1) {
            ans += (unsigned long long)vs[hRow].size() - (unsigned long long)col;
        } else {
            unsigned long long lp = 1, rp = len;
            while (lp <= rp) {
                ans += lp * rp * (lp == rp ? 1ULL : 2ULL);
                lp++; rp--;
            }
            bins(col + 1, lRow, hRow);
        }
    }
}
0