QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#718171#9406. TriangleforgotmyhandleWA 30ms96016kbC++147.0kb2024-11-06 19:52:532024-11-06 19:52:55

Judging History

你现在查看的是最新测评结果

  • [2024-11-06 19:52:55]
  • 评测
  • 测评结果:WA
  • 用时:30ms
  • 内存:96016kb
  • [2024-11-06 19:52:53]
  • 提交

answer

#include <iostream>
#include <unordered_map>
#include <algorithm>
#include <string.h>
#include <cassert>
#include <map>
#define lowbit(x) ((x) & (-(x)))
#define int long long
using namespace std;
const int B = 13331;
const int P = 1000000007;
int n, m;
string str[300005], s[300005];
int sa[600005], rk[600005], ht[600005];
pair<int, int> mn[20][600005];
int lg2[600005];
namespace SA {
    int c[600005], x[600005], y[600005];
    int t[600005];
    void SA(string str, int n) {
        int m = 600000;
        for (int i = 0; i <= m; i++) c[i] = 0;
        for (int i = 1, sep = 255; i <= n; i++) c[x[i] = t[i] = (str[i] == '/' ? ++sep : str[i])]++;
        for (int i = 1; i <= m; i++) c[i] += c[i - 1];
        for (int i = 1; i <= n; i++) sa[c[x[i]]--] = i;
        for (int d = 1; d <= n; d <<= 1) {
            int tmp = 0;
            for (int i = n - d + 1; i <= n; i++) y[++tmp] = i;
            for (int i = 1; i <= n; i++) {
                if (sa[i] > d) 
                    y[++tmp] = sa[i] - d;
            }
            for (int i = 1; i <= m; i++) c[i] = 0;
            for (int i = 1; i <= n; i++) c[x[i]]++;
            for (int i = 1; i <= m; i++) c[i] += c[i - 1];
            for (int i = n; i; i--) sa[c[x[y[i]]]--] = y[i], y[i] = 0;
            swap(x, y);
            x[sa[1]] = tmp = 1;
            for (int i = 2; i <= n; i++) 
                x[sa[i]] = (tmp += (y[sa[i]] != y[sa[i - 1]] || y[sa[i] + d] != y[sa[i - 1] + d]));
            m = tmp;
            if (tmp == n) 
                break;
        }
        t[n + 1] = -1;
        for (int i = 1; i <= n; i++) rk[sa[i]] = i;
        for (int i = 1, k = 0; i <= n; i++) {
            k -= (k != 0);
            while (t[i + k] == t[sa[rk[i] - 1] + k]) ++k;
            ht[rk[i]] = k;
        }
        lg2[0] = -1;
        for (int i = 1; i <= n; i++) mn[0][i] = make_pair(ht[i], i), lg2[i] = lg2[i - 1] + ((i & (i - 1)) == 0);
        for (int i = 1; (1 << i) <= n; i++) {
            for (int j = 1; j + (1 << i) - 1 <= n; j++) 
                mn[i][j] = min(mn[i - 1][j], mn[i - 1][j + (1 << (i - 1))]);
        }
    }
}
int bg[300005], fdsa[300005];
int len[300005], o[300005], cnt[300005];
inline pair<int, int> Query(int l, int r) {
    assert(l <= r);
    int k = lg2[r - l + 1];
    return min(mn[k][l], mn[k][r - (1 << k) + 1]);
}
inline int lcp(int x, int y) { return Query(min(rk[x], rk[y]) + 1, max(rk[x], rk[y])).first; }
// lcp of str[x]'s p suffix and str[y]'s q suffix
inline int LCP(int x, int y, int p = 1, int q = 1) { return lcp(bg[x] + p - 1, bg[y] + q - 1); }
// return whether (str[x]'s p suffix) is lexicographically strictly smaller than (str[y]'s q suffix)
bool cmp(int x, int y, int p = 1, int q = 1) {
    int a = LCP(x, y, p, q);
    // cerr << x << " " << y << " " << a << " " << len[x] << " " << len[y] << "\n";
    assert(p + a - 1 <= len[x] && q + a - 1 <= len[y]);
    if (p + a > len[x] && q + a > len[y]) 
        return 0;
    if (p + a > len[x]) 
        return 1;
    if (q + a > len[y]) 
        return 0;
    return str[x][p + a] < str[y][q + a];
}
map<int, int> mp;
int ans, acnt;
int app[300005], pre[300005];
int H[300005], val[300005];
struct qquery {
    int x, l, r, v;
} qs[300005];
int qcnt;
struct BIT {
    int bit[300005];
    void add(int x, int y) { for (; x <= 300000; x += lowbit(x)) bit[x] += y; }
    void add(int l, int r, int v) { add(l, v), add(r + 1, -v); }
    int query(int x) {
        int ret = 0;
        for (; x; x -= lowbit(x)) ret += bit[x];
        return ret;
    }
} bit;
signed main() {
    int tc;
    cin >> tc;
    while (tc--) {
        mp.clear();
        cin >> n;
        string S = " ";
        for (int i = 1; i <= n; i++) cin >> str[o[i] = i], fdsa[i] = bg[i] = S.size(), S += str[i], S += '/', len[i] = str[i].size();
        // cout << S << "\n";
        // for (int i = 1; i <= n; i++) cout << bg[i] << " ";
        // cout << "\n";
        SA::SA(S, (int)S.size() - 1);
        // cerr << LCP(3, 2) << "\n";
        for (int i = 1; i <= n; i++) s[i] = str[i] = ' ' + str[i];
        sort(o + 1, o + n + 1, [](int a, int b) { return cmp(a, b); });
        // return 0;
        m = n, n = ans = 0;
        for (int i = 1; i <= m; i++) {
            if (i == 1 || s[o[i]] != s[o[i - 1]]) {
                cnt[++n] = 1;
                bg[n] = fdsa[o[i]], str[n] = s[o[i]];
            } else 
                ++cnt[n];
        }
        for (int i = 1; i <= n; i++) len[i] = (int)str[i].size() - 1, pre[i] = pre[i - 1] + cnt[i];
        // cout << n << "\n";
        // for (int i = 1; i <= n; i++) cout << str[i] << " " << bg[i] << " " << len[i] << " " << cnt[i] << "\n";
        // 3 sames
        for (int i = 1; i <= n; i++) ans += cnt[i] * (cnt[i] - 1) * (cnt[i] - 2) / 6;

        // 2 sames
        for (int i = 1; i <= n; i++) ans += pre[i - 1] * cnt[i] * (cnt[i] - 1) / 2;

        // no sames
        for (int i = 1, &h = H[i], tmp; i <= n; i++) {
            for (int j = 1 + (tmp = h = acnt = qcnt = 0); j <= len[i]; j++) {
                h = h * B + str[i][j] - 'a' + 1;
                if (mp.count(h)) {
                    app[++acnt] = mp[h];
                    if (cmp(i, app[acnt], len[app[acnt]] + 1)) 
                        tmp += cnt[app[acnt]] * (cnt[app[acnt]] - 1) / 2;
                }
            }
            mp[h] = i;
            for (int j = 1; j <= acnt; j++) {
                int l = 1, r = app[j] - 1, mid, asdf = app[j];
                while (l <= r) {
                    mid = (l + r) >> 1;
                    if (cmp(i, mid, len[app[j]] + 1)) 
                        asdf = mid, r = mid - 1;
                    else 
                        l = mid + 1;
                }
                l = asdf, r = app[j] - 1;
                if (l <= r) {
                    tmp += (pre[r] - pre[l - 1]) * cnt[app[j]];
                    qs[++qcnt] = (qquery) { app[j], l, r, cnt[j] };
                }

                val[j] = i, l = app[j] + 1, r = i - 1;
                while (l <= r) {
                    mid = (l + r) >> 1;
                    if (cmp(i, mid, len[app[j]] + 1)) 
                        val[j] = mid, r = mid - 1;
                    else 
                        l = mid + 1;
                }
                qs[++qcnt] = (qquery) { val[j], app[j], -1, cnt[j] };
                tmp += (pre[i - 1] - pre[val[j] - 1]) * cnt[app[j]];
            }
            sort(qs + 1, qs + qcnt + 1, [](qquery a, qquery b) { return (a.x == b.x) ? (a.r < b.r) : (a.x < b.x); });
            for (int j = qcnt; j; j--) {
                if (qs[j].r != -1) 
                    bit.add(qs[j].l, qs[j].r, qs[j].v);
                else 
                    tmp -= qs[j].v * bit.query(qs[j].l);
            }
            for (int j = 1; j <= qcnt; j++) qs[j].r != -1 ? bit.add(qs[j].l, qs[j].r, -qs[j].v) : void();
            ans += tmp * cnt[i];
        }
        cout << ans << "\n";
    }
    return 0;
}
/*
3
6
cbaa
cb
cb
cbaa
ba
ba
6
cbaa
cb
cb
cbaa
ba
ba
6
cbaa
cb
cb
cbaa
ba
ba
*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 4ms
memory: 76072kb

input:

3
6
cbaa
cb
cb
cbaa
ba
ba
3
sdcpc
sd
cpc
1
ccpc

output:

16
0
0

result:

ok 3 lines

Test #2:

score: 0
Accepted
time: 28ms
memory: 86448kb

input:

14
1
lfpbavjsm
2
pdtlkfwn
mbd
3
dvqksg
dvqksg
uhbhpyhj
4
ombwb
ombwb
ombwb
ombwb
5
oztclz
oztclz
oztclz
oztclz
kul
6
vco
vco
vco
dtktsqtinm
vco
vco
7
tb
tb
kowbsu
ygkfphcij
tb
uvei
tb
8
vxxtxssht
abnsxbf
bydaae
bydaae
udalyvmcef
bydaae
bydaae
bydaae
9
aaze
zvyonw
qjfv
mmhkef
qjfv
qjfv
qjfv
mmhkef
qj...

output:

0
0
0
4
10
20
10
20
41
14
63
74
18
11081

result:

ok 14 lines

Test #3:

score: 0
Accepted
time: 30ms
memory: 88940kb

input:

11
10
bppzfsncq
bppzfsncq
vcqxgcehdx
bppzfsncq
bppzfsncq
muwrcvt
w
aanwhqmync
aanwhqmync
bppzfsncq
10
t
jkky
t
z
t
t
z
z
z
t
10
qidkyca
uhqubvbo
kosyvh
gsj
gsj
gsj
duo
jrro
gsj
jrro
10
lhktb
lhktb
lhktb
uohl
lhktb
r
lhktb
lhktb
wruim
lhktb
10
e
gqvdmpvxb
gqvdmpvxb
gqvdmpvxb
sttirbhz
gqvdmpvxb
zdfpm
...

output:

30
60
15
35
20
20
23
12
38
44
8047

result:

ok 11 lines

Test #4:

score: -100
Wrong Answer
time: 28ms
memory: 96016kb

input:

11
100
kalgqjh
mdszzwe
qxn
kalgqjh
hy
kalgqjh
suplvp
r
kkeoxmx
tcoise
suplvp
suplvp
y
kalgqjh
vrwniyici
jmnyrradyq
kalgqjh
kalgqjh
suplvp
rkg
xzevyk
zc
suplvp
hcupv
kalgqjh
qakyahjaoi
mum
pbg
u
ip
kalgqjh
kalgqjh
jngc
ylr
suplvp
qxn
kalgqjh
bzwodm
e
kalgqjh
kalgqjh
evmm
kbymvbccs
kalgqjh
suplvp
kalg...

output:

12478
6722
9220
6668
4934
11233
7950
5470
4525
5743
1586184

result:

wrong answer 11th lines differ - expected: '1586066', found: '1586184'