题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1590
题意：给出 n 条信息，m 条密码，问对于每一条密码，它是多少个信息的前缀或多少个信息是它的前缀
直接对于信息构造字典树，每条信息的终止节点权值加一，这样读入密码的时候走一遍树上路径即可知道多少个信息是它的前缀，它是多少个前缀怎么求呢？记录每个节点字数权值大小即可，答案加上密码终止节点子树权值，但是有一个问题是这样的话如果两个串正好相同会被统计两次，再减去这个点的权值即可。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
#include <cmath> #include <queue> #include <cstdio> #include <iomanip> #include <cstdlib> #include <cstring> #include <iostream> #include <algorithm> #define N 500010 #define ll long long using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int ch[N][2],w[N],sz[N],tot=1; void add() { int len=read(),now=1; for(int i=1;i<=len;i++) { int x=read(); if(!ch[now][x]) ch[now][x]=++tot; now=ch[now][x]; } w[now]++; } void get_sz(int x) { sz[x]=w[x]; if(ch[x][0]) { get_sz(ch[x][0]); sz[x]+=sz[ch[x][0]]; } if(ch[x][1]) { get_sz(ch[x][1]); sz[x]+=sz[ch[x][1]]; } } int n,m; int main() { n=read(),m=read(); for(int i=1;i<=n;i++) add(); get_sz(1); for(int i=1;i<=m;i++) { int len=read(),ans=0,now=1; for(int j=1;j<=len;j++) { int x=read(); now=ch[now][x]; ans+=w[now]; } ans+=sz[now]-w[now]; printf("%d\n",ans); } } |
Comments | 2 条评论
写的很棒棒，学到了很多
大佬…