Stilwell正在玩一款名为Danganronpa的推理类射击游戏,在这款游戏中,玩家需要选择对应的线索作为子弹去射击有漏洞的证言来完成反驳。
先给出 言弹 和 证言 的定义:
言弹 :用于反驳的言论子弹,由一个字符串 Ai 表示。
证言 :由剧情发展而产生的一些言论,言论中可能存在破绽与矛盾,由一个字符串 Bi 表示。
伤害 :用 言弹 A 射击 证言 B 造成的 伤害值 由函数 F(A,B) 表示, F(A,B) 的值为串 A 在串 B 中的 出现次数 。
由于Stilwell并不擅长推理,所以每次都会用同一言弹对所有证言进行射击。
游戏进行到后期,流程将会变得复杂一些, 证言 和 言弹 都会不断增加,具体的情形可以这样描述:
INSERT c v :增加一枚新 言弹 ,选择一个没有使用过的最小标号 u ,并使 Au=Av+c (此处的保证为单个字符,加法操作表示把字符 c 接在原串后面,下同)。
ADD c v :增加一句新 证言 ,选择一个没有使用过的最小标号 u (言弹和证言的标号的分开的),并使 Bu=Bv+c 。
SHOOT u(u>0) :发射一枚编号为 u 的 言弹 ,对 当前存在的所有证言 造成伤害(对每个证言的伤害是单独计算的,即对第 i 句证言造成 F(Au,Bi) 的伤害)。
QUERY u(u>0) :询问第 u 句 证言 当前总共受到了多少伤害(伤害从这句证言被加入之后开始计算)。
A0 和 B0 是空串, c 为小写字母。
先给出 言弹 和 证言 的定义:
言弹 :用于反驳的言论子弹,由一个字符串 Ai 表示。
证言 :由剧情发展而产生的一些言论,言论中可能存在破绽与矛盾,由一个字符串 Bi 表示。
伤害 :用 言弹 A 射击 证言 B 造成的 伤害值 由函数 F(A,B) 表示, F(A,B) 的值为串 A 在串 B 中的 出现次数 。
由于Stilwell并不擅长推理,所以每次都会用同一言弹对所有证言进行射击。
游戏进行到后期,流程将会变得复杂一些, 证言 和 言弹 都会不断增加,具体的情形可以这样描述:
INSERT c v :增加一枚新 言弹 ,选择一个没有使用过的最小标号 u ,并使 Au=Av+c (此处的保证为单个字符,加法操作表示把字符 c 接在原串后面,下同)。
ADD c v :增加一句新 证言 ,选择一个没有使用过的最小标号 u (言弹和证言的标号的分开的),并使 Bu=Bv+c 。
SHOOT u(u>0) :发射一枚编号为 u 的 言弹 ,对 当前存在的所有证言 造成伤害(对每个证言的伤害是单独计算的,即对第 i 句证言造成 F(Au,Bi) 的伤害)。
QUERY u(u>0) :询问第 u 句 证言 当前总共受到了多少伤害(伤害从这句证言被加入之后开始计算)。
A0 和 B0 是空串, c 为小写字母。
考虑每枚言弹对所有证言造成的伤害,询问时用当前受到的总伤害减去出现之前受到的总伤害即可
对于INSERT和ADD操作增加的字符,存在一棵trie上,对它造一个sa
实现可以用波兰表储存,方便计算出height数组
有了height数组,一枚言弹打出时能造成伤害的串一定是一段连续的区间
在sa上用单调栈 + 二分就可以求出这个区间的左右端点
对于一个串,它被言弹攻击到的次数,等于代表它的节点受到的攻击的次数加上其每个祖先受到的攻击次数
那么对于trie求个dfs括号序,左括号权值为1,右括号权值为-1
在dfs序上,一个点的左括号位置起的前缀和就是这个点到目前受到的攻击次数
那么对于每个点,以rank为第一关键字,dfn为第二关键字
言弹攻击时等于对rank属于某个区间的点造成伤害
查询时等于是对dfn不超过某值的点查询前缀和
这些都可以用k-d tree实现,总的算法复杂度为O(nlogn + nsqrt(n))
实践证明k-d tree使用方差剖分的效果比按x、y轴交替剖分的效果好一些
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
#include<stack>
#include<ctime>
#define min(a,b) ((a) < (b) ? (a) : (b))
#define max(a,b) ((a) > (b) ? (a) : (b))
using namespace std;
const int N = 17;
const int INF = ~0U>>1;
const int maxn = 2E5 + 20;
const int maxm = maxn * 2;
typedef double DB;
typedef long long LL;
struct data{
int typ,Num; data(){}
data(int typ,int Num): typ(typ),Num(Num){}
}D[maxm];
struct Point{
int x,y,typ; Point(){}
Point(int x,int y,int typ): x(x),y(y),typ(typ){}
}p[maxm];
int m,cnt,tpA,tpB,tp,rt,stp,dfs_clock,lastA[maxn],lastB[maxn],ch[maxn][26],
fa[maxn][N],L[maxn],Rank[maxn][N],Sa[maxn],Height[maxn],c[maxn],t[maxn],
Left[maxn],Right[maxn],mx[maxm][2],my[maxm][2],lp[maxn],rp[maxn],stk[maxn],
cov[maxn],Add[maxm],siz[maxm],son[maxm][2],Key[maxm];
char s[20];
LL delta[maxn],tot[maxm];
vector <int> g[maxn];
inline bool cmpx(const Point &A,const Point &B) {return A.x < B.x;}
inline bool cmpy(const Point &A,const Point &B) {return A.y < B.y;}
inline int getint()
{
char ch = getchar(); int ret = 0;
while (ch < '0' || '9' < ch) ch = getchar();
while ('0' <= ch && ch <= '9')
ret = ret * 10 + ch - '0',ch = getchar();
return ret;
}
inline int getcom()
{
scanf("%s",s + 1);
if (s[1] == 'I') return 1;
if (s[1] == 'A') return 2;
if (s[1] == 'S') return 3; return 4;
}
inline int Get()
{
char ch = getchar();
while (ch < 'a' || 'z' < ch) ch = getchar();
return ch;
}
inline int Search(int key)
{
int l = 1,r = stp + 1; stk[r] = 0;
while (r - l > 1)
{
int mid = l + r >> 1;
if (Height[stk[mid]] < key) l = mid; else r = mid;
}
return Height[stk[l]] >= key ? l : r;
}
void Pre_Work()
{
m = getint();
for (int i = 1; i <= m; i++)
{
int typ = getcom(),now,Nex,Num;
if (typ == 1)
{
Nex = Get() - 'a',Num = lastA[getint()];
if (!ch[Num][Nex])
{
ch[Num][Nex] = ++cnt; Rank[cnt][0] = Nex + 1;
fa[cnt][0] = Num; L[cnt] = L[Num] + 1;
}
lastA[++tpA] = ch[Num][Nex];
}
else if (typ == 2)
{
Nex = Get() - 'a',Num = lastB[getint()];
if (!ch[Num][Nex])
{
ch[Num][Nex] = ++cnt; Rank[cnt][0] = Nex + 1;
fa[cnt][0] = Num; L[cnt] = L[Num] + 1;
}
lastB[++tpB] = ch[Num][Nex]; ++cov[ch[Num][Nex]]; D[++tp] = data(3,tpB);
}
else if (typ == 3) now = getint(),D[++tp] = data(1,now);
else if (typ == 4) now = getint(),D[++tp] = data(2,now);
}
}
void Build()
{
int n = 26,*y = t;
for (int i = 1; i <= cnt; i++) ++c[Rank[i][0]];
for (int i = 2; i <= n; i++) c[i] += c[i - 1];
for (int i = 1; i <= cnt; i++) Sa[c[Rank[i][0]]--] = i;
for (int k = 1; k < N; k++)
{
int p = 0;
for (int i = 1; i <= cnt; i++)
{
fa[i][k] = fa[fa[i][k - 1]][k - 1];
if (!fa[i][k - 1]) y[++p] = i;
else g[fa[i][k - 1]].push_back(i);
}
for (int i = 1; i <= cnt; i++)
{
for (int j = 0; j < g[Sa[i]].size(); j++)
y[++p] = g[Sa[i]][j];
g[Sa[i]].clear();
}
for (int i = 1; i <= n; i++) c[i] = 0;
for (int i = 1; i <= cnt; i++) ++c[Rank[i][k - 1]];
for (int i = 2; i <= n; i++) c[i] += c[i - 1];
for (int i = cnt; i; i--) Sa[c[Rank[y[i]][k - 1]]--] = y[i];
Rank[Sa[1]][k] = p = 1;
for (int i = 2; i <= cnt; i++)
Rank[Sa[i]][k] = Rank[Sa[i]][k - 1] == Rank[Sa[i - 1]][k - 1] &&
Rank[fa[Sa[i]][k - 1]][k - 1] == Rank[fa[Sa[i - 1]][k - 1]][k - 1] ? p : ++p;
n = p;
if (n == cnt)
{
for (int i = k + 1; i < N; i++)
for (int j = 1; j <= cnt; j++)
Rank[j][i] = Rank[j][i - 1];
break;
}
}
for (int i = 2; i <= cnt; i++)
{
int A = Sa[i - 1],B = Sa[i];
for (int j = N - 1; j >= 0; j--)
if (Rank[A][j] == Rank[B][j])
Height[i] += (1 << j),A = fa[A][j],B = fa[B][j];
}
}
inline void Dfs(int x)
{
Left[x] = ++dfs_clock; p[dfs_clock] = Point(Rank[x][N - 1],dfs_clock,cov[x]);
for (int i = 0; i < 26; i++) if (ch[x][i]) Dfs(ch[x][i]);
Right[x] = ++dfs_clock; p[dfs_clock] = Point(Rank[x][N - 1],dfs_clock,-cov[x]);
}
void LP_RP()
{
Height[0] = INF; stk[stp = 1] = lp[Sa[1]] = 1;
for (int i = 2; i <= cnt; i++)
{
while (stp && Height[i] <= Height[stk[stp]])
--stp; stk[++stp] = i;
int pos = Search(L[Sa[i]]);
if (pos == 1) lp[Sa[i]] = 1;
else if (pos > stp) lp[Sa[i]] = Rank[Sa[i]][N - 1];
else lp[Sa[i]] = stk[pos - 1];
}
stk[stp = 1] = rp[Sa[cnt]] = cnt;
for (int i = cnt - 1; i; i--)
{
int pos = Search(L[Sa[i]]);
if (pos == 1) rp[Sa[i]] = cnt;
else if (pos > stp) rp[Sa[i]] = Rank[Sa[i]][N - 1];
else rp[Sa[i]] = stk[pos - 1] - 1;
while (stp && Height[i] <= Height[stk[stp]])
--stp; stk[++stp] = i;
}
}
inline int KD_Build(int l,int r)
{
if (l > r) return 0;
if (l == r)
{
mx[l][0] = mx[l][1] = p[l].x;
my[l][0] = my[l][1] = p[l].y;
siz[l] = p[l].typ; return l;
}
int mid = l + r >> 1;
DB avx,avy,sumx,sumy;
avx = avy = sumx = sumy = 0;
for (int i = l; i <= r; i++) avx += (DB)(p[i].x),avy += (DB)(p[i].y);
avx /= (DB)(r - l + 1); avy /= (DB)(r - l + 1);
for (int i = l; i <= r; i++)
{
DB K = (DB)(p[i].x) - avx; sumx += K * K;
K = (DB)(p[i].y) - avy; sumy += K * K;
}
if (sumx < sumy) nth_element(p + l,p + mid,p + r + 1,cmpy);
else nth_element(p + l,p + mid,p + r + 1,cmpx); siz[mid] = p[mid].typ;
mx[mid][0] = mx[mid][1] = p[mid].x; son[mid][0] = KD_Build(l,mid - 1);
my[mid][0] = my[mid][1] = p[mid].y; son[mid][1] = KD_Build(mid + 1,r);
if (son[mid][0])
{
siz[mid] += siz[son[mid][0]];
mx[mid][0] = min(mx[mid][0],mx[son[mid][0]][0]);
mx[mid][1] = max(mx[mid][1],mx[son[mid][0]][1]);
my[mid][0] = min(my[mid][0],my[son[mid][0]][0]);
my[mid][1] = max(my[mid][1],my[son[mid][0]][1]);
}
if (son[mid][1])
{
siz[mid] += siz[son[mid][1]];
mx[mid][0] = min(mx[mid][0],mx[son[mid][1]][0]);
mx[mid][1] = max(mx[mid][1],mx[son[mid][1]][1]);
my[mid][0] = min(my[mid][0],my[son[mid][1]][0]);
my[mid][1] = max(my[mid][1],my[son[mid][1]][1]);
}
return mid;
}
inline void pushdown(int x)
{
if (!Add[x]) return;
Key[x] += p[x].typ * Add[x];
tot[x] += 1LL * siz[x] * Add[x];
Add[son[x][0]] += Add[x];
Add[son[x][1]] += Add[x]; Add[x] = 0;
}
inline void KD_Add(int x,int lx,int rx)
{
if (lx <= mx[x][0] && mx[x][1] <= rx)
{
++Add[x]; pushdown(x); return;
}
pushdown(x); //++Nodes;
if (mx[x][1] < lx || rx < mx[x][0]) return;
KD_Add(son[x][0],lx,rx); KD_Add(son[x][1],lx,rx);
if (lx <= p[x].x && p[x].x <= rx) Key[x] += p[x].typ;
tot[x] = tot[son[x][0]] + tot[son[x][1]] + 1LL * Key[x];
}
inline LL KD_Query(int x,int pos)
{
if (!x) return 0; pushdown(x); //++Nodes;
if (my[x][1] <= pos) return tot[x];
if (pos < my[x][0]) return 0;
LL now = p[x].y <= pos ? Key[x] : 0;
return now + KD_Query(son[x][0],pos) + KD_Query(son[x][1],pos);
}
void Print()
{
for (int i = 0; i <= cnt; i++)
for (int j = 0; j < 26; j++)
if (ch[i][j]) printf("%d %d %c\n",i,ch[i][j],j + 'a');
}
void Printf(LL x)
{
if (!x) {puts("0"); return;} int len = 0;
while (x) s[++len] = x % 10LL,x /= 10LL;
for (int i = len; i; i--) putchar(s[i] + '0'); puts("");
}
int main()
{
#ifdef DMC
freopen("DMC.txt","r",stdin);
freopen("test.txt","w",stdout);
#endif
Pre_Work(); Build(); Dfs(0); //Print();
//cerr << (double)(clock()) / CLOCKS_PER_SEC << endl;
rt = KD_Build(2,dfs_clock - 1); LP_RP();
//cerr << (double)(clock()) / CLOCKS_PER_SEC << endl;
for (int i = 1; i <= tp; i++)
if (D[i].typ == 1) KD_Add(rt,lp[lastA[D[i].Num]],rp[lastA[D[i].Num]]);
else if (D[i].typ == 2)
Printf(KD_Query(rt,Left[lastB[D[i].Num]]) - delta[lastB[D[i].Num]]);
else delta[lastB[D[i].Num]] = KD_Query(rt,Left[lastB[D[i].Num]]);
//cout << Nodes << endl;
//cerr << (double)(clock()) / CLOCKS_PER_SEC << endl;
return 0;
}