[SAM模板题] P3975 [TJOI2015] 弦论

题目思路

对原串建立SAMSAMSAM，然后分情况考虑：

t=0t = 0t=0，代表本质不同即为不同，那么我们需要对每个SAMSAMSAM节点uuu维护uuu能够到达的点的数量，即为经过uuu的本质不同的子串的数量。
t=1t = 1t=1，代表位置不同即为不同，那么我们考虑对于每个节点uuu计算经过该点位置不同的子串数量。那么我们可以想到这需要先求串sss的出现次数，也就是endpos(s)endpos(s)endpos(s)的大小，而根到uuu表示的串的endposendposendpos大小为parenttreeparent\ treeparent tree上uuu的子树中的前缀点的数量

不难发现，我们可以维护一个dp[u]dp[u]dp[u]表示不同限制下的子串数目，以及每个点的点权。对于t=0t = 0t=0，所有点的点权都是111，否则先dfsdfsdfs求子树大小作为点权，然后dfsdfsdfs求答案即可。

#include <bits/stdc++.h>
#pragma gcc optimize("O2")
#pragma g++ optimize("O2")
#define int long long
#define endl '\n'
using namespace std;const int N = 1e6 + 10, MOD = 1e9 + 7;int dp[N], siz[N];struct SAM{int ch[N << 1][26], fa[N << 1], len[N << 1], vis[N << 1];int last, tot;SAM(): last(1), tot(1) {}inline void extend(int x){ //*单字符扩展int p = last, np = last = ++tot;len[np] = len[p] + 1, vis[np] = dp[np] = siz[np] = 1;for(; p && !ch[p][x]; p = fa[p]) ch[p][x] = np;if(!p) fa[np] = 1;else{int q = ch[p][x];if(len[q] == len[p] + 1) fa[np] = q;else {int nq = ++tot;memcpy(ch[nq], ch[q], sizeof(ch[nq]));fa[nq] = fa[q], fa[np] = fa[q] = nq, len[nq] = len[p] + 1;for(; ch[p][x] == q; p = fa[p]) ch[p][x] = nq;}}}
}sam;vector<int> g[N];void dfs1(int u){for(auto v : g[u]){dfs1(v);siz[u] += siz[v];}dp[u] = siz[u];
}bool vis[N];int dfs2(int u){if(vis[u]) return dp[u];vis[u] = true;for(int i = 0; i < 26; i++){int v = sam.ch[u][i];if(v) dp[u] += dfs2(v);}return dp[u];
}void build(int t){for(int i = 1; i <= sam.tot; i++){if(sam.fa[i]) g[sam.fa[i]].emplace_back(i);}if(!t){for(int i = 1; i <= sam.tot; i++) dp[i] = siz[i] = 1;} else {dfs1(1);}dp[1] = siz[1] = 0;dfs2(1);
}void query(int u, int k){if(k > dp[u]){cout << -1;return;}if(k <= siz[u]) return;k -= siz[u];for(int i = 0; i < 26; i++){int v = sam.ch[u][i];if(k > dp[v]) k -= dp[v];else{cout << (char)('a' + i);query(v, k);return;}}
}inline void solve(){string s; cin >> s;for(int i = 0; i < s.size(); i++) sam.extend(s[i] - 'a');int t, k; cin >> t >> k;build(t);query(1, k);
}signed main(){//ios_base::sync_with_stdio(false), cin.tie(0);cout << fixed << setprecision(12);int t = 1; //cin >> t;while(t--) solve();return 0;
}

[SAM模板题] P3975 [TJOI2015] 弦论相关推荐

luogu P3975 [TJOI2015]弦论 SAM
luogu P3975 [TJOI2015]弦论链接 bzoj 思路建出sam. 子串算多个的,统计preant tree的子树大小,否则就是大小为1 然后再统计sam的节点能走到多少串. 然后就 ...
洛谷 P3975 [TJOI2015]弦论解题报告
P3975 [TJOI2015]弦论题目描述为了提高智商,ZJY开始学习弦论.这一天,她在<String theory>中看到了这样一道问题:对于一个给定的长度为\(n\)的字符串,求 ...
洛谷 [P3975 [TJOI2015]弦论
洛谷 P3975 [TJOI2015]弦论题目描述给定一个长度为 nnn 的字符串,求它的第 kkk 小字串:给定 ttt, ttt 为 000 则表示不同位置的相同子串算作一个,ttt 为 11 ...
P3975 [TJOI2015]弦论（SAM）
[TJOI2015]弦论 - 洛谷感觉位置不同算相同还是比较好想的,我们top排序后算每个点后面连接了多少点就知道里面有多少字串了,然后求k大就行了: 考了位置不同算不同的话,我们就要考虑endpo ...
【后缀自动机】Luogu P3975 [TJOI2015]弦论题解
[TJOI2015]弦论题目描述为了提高智商,ZJY 开始学习弦论.这一天,她在<String theory>中看到了这样一道问题:对于一个给定的长度为 nnn 的字符串,求出它的第 ...
洛谷P3975 - [TJOI2015]弦论
Portal Description 给出一个小写字母串\(s(|s|\leq5\times10^5),t\in\{0,1\},k(k\leq10^9)\),求\(s\)的第\(k\)小子串.\(t= ...
P3975 [TJOI2015]弦论 - 后缀自动机（SAM）
这是一道板子题的改编,意在加深对求第k小子串的理解.首先先看一下最简单的SAM板子.相信应该都写过了才会写这题 // // Created by acer on 2021/2/16. // //判断子 ...
[Luogu P3975] [TJOI2015]弦论
洛谷传送门 BZOJ传送门题目描述为了提高智商,ZJY开始学习弦论.这一天,她在< String theory>中看到了这样一道问题:对于一个给定的长度为nnn的字符串,求出它的第k& ...
P3975 [TJOI2015]弦论第K小子串
题目描述 https://www.luogu.org/problem/P3975 为了提高智商,ZJY开始学习弦论.这一天,她在< String theory>中看到了这样一道问题:对于一 ...

[SAM模板题] P3975 [TJOI2015] 弦论

题目思路

[SAM模板题] P3975 [TJOI2015] 弦论相关推荐

最新文章

热门文章