結果

問題 No.430 文字列検索
コンテスト
ユーザー fgwiebfaoish
提出日時 2022-09-11 14:49:21
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 80 ms / 2,000 ms
コード長 11,425 bytes
コンパイル時間 3,270 ms
コンパイル使用メモリ 114,432 KB
実行使用メモリ 30,336 KB
最終ジャッジ日時 2024-11-10 01:01:43
合計ジャッジ時間 2,627 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.Collections.Generic;
using System.Collections;
using System.Collections.Specialized;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.IO;
using System.Reflection;
using static System.Math;
using System.Numerics;
using System.Threading;
using System.Runtime.CompilerServices;
using System.Diagnostics;
static class Program{
	const long inf=long.MaxValue>>1;
	const int mod=998244353;
	static Sc sc=new Sc();
	static void Main(){
		Pt pt=new Pt(Solve);
		pt.Sm();
	}
	static void Solve(Pt pt) {
		string s=sc.S;
		int n=sc.I;
		var t=sc.Arr(n,()=>sc.S);
		var ac=new Ac<Stl3>(t,'A','Z');
		long ans=0;
		ac.Match(s,false,(a,e)=>{ans+=e.cnt;});
		pt.Wl(ans);
	}
}
public class Ac<T> where T:Stl,new(){
	static private int o;
	public int n,cnt=1;
	public Nd root;
	public Nd[] h;
	public Nd[] q;
	public class Nd{
		public Nd[] ar;
		public List<Nd> li=new List<Nd>();
		public Nd sx;
		public T stl=new T();
		public int c,f=0,n;
		public bool b=false;
		public Nd(int c,int f){this.c=c-o;this.f=f;}
		public int cnt{get{return stl.cnt;}}
	}
	public Ac(string[] t,int a,int z){
		new T().Set(t);
		n=z-a+1;
		o=a;
		root=new Nd(a-1,0);
		q=new Nd[t.Length];
		for(int i = 0;i<t.Length;i++) {
			var r=root;
			for(int j = 0;j<t[i].Length;j++) {
				if(!r.b){r.ar=new Nd[n];r.b=true;}
				if(r.ar[t[i][j]-o]==null){r.li.Add(r=r.ar[t[i][j]-o]=new Nd(t[i][j],j+1));cnt++;}
				else{r=r.ar[t[i][j]-o];}
			}
			q[i]=r;
			r.stl.Add(i);
		}
		h=new Nd[cnt];
		h[0]=root;
		int p=1;
		var qu=new Queue<Nd>();
		for(int i = 0;i<root.li.Count;i++,p++) {
			root.li[i].sx=root;
			if(root.li[i].b){qu.Enqueue(root.li[i]);}
			root.li[i].n=p;
			h[p]=root.li[i];
		}
		while(qu.Count>0){
			var e=qu.Dequeue();
			for(int i = 0;i<e.li.Count;i++,p++) {
				var r=e.sx;
				while(r!=null&&(!r.b||r.ar[e.li[i].c]==null)){r=r.sx;}
				e.li[i].sx=r!=null?r.ar[e.li[i].c]:root;
				if(e.li[i].sx.stl.exist){e.li[i].stl.AddRange(e.li[i].sx.stl);}
				if(e.li[i].b){qu.Enqueue(e.li[i]);}
				e.li[i].n=p;
				h[p]=e.li[i];
			}
		}
		if(cnt==1){
			root.ar=new Nd[n];
			for(int i = 0;i<root.ar.Length;i++) {root.ar[i]=root;}
		}
	}
	public void Match(string t,bool bo,Action<int,Nd> f){
		var e=root;
		for(int i = 0;i<t.Length;i++) {
			while(e!=root&&(!e.b||e.ar[t[i]-o]==null)){e=e.sx;}
			if(e.ar[t[i]-o]!=null){e=e.ar[t[i]-o];}
			if(bo||e.stl.exist){f(i,e);}
		}
	}
	public void Match(string t,bool bo,Func<int,Nd,bool> f){
		var e=root;
		for(int i = 0;i<t.Length;i++) {
			while(e!=root&&(!e.b||e.ar[t[i]-o]==null)){e=e.sx;}
			if(e.ar[t[i]-o]!=null){e=e.ar[t[i]-o];}
			if(bo||e.stl.exist){
				if(!f(i,e)){break;}
			}
		}
	}
	public Nd Av(Nd e,char c){
		while(e!=root&&(!e.b||e.ar[c-o]==null)){e=e.sx;}
		if(e.ar[c-o]!=null){e=e.ar[c-o];}
		return e;
	}
}
public interface Stl {
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	void Add(int x);
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	void AddRange(Stl x);
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	void Set(string[] t);
	bool exist{get;}
	int cnt{get;}
}
public class Stl1:Stl{
	static private string[] t;
	public List<string> dt=new List<string>();
	public void Add(int x)=>dt.Add(t[x]);
	public void AddRange(Stl x)=>dt.AddRange(((Stl1)x).dt);
	public void Set(string[] t)=>Stl1.t=t;
	public bool exist{get{return dt.Count!=0;}}
	public int cnt{get{return dt.Count;}}
	public override string ToString(){return string.Join(" ",dt);}
}
public class Stl2:Stl{
	static private string[] t;
	public List<int> dt=new List<int>();
	public void Add(int x)=>dt.Add(t[x].Length);
	public void AddRange(Stl x)=>dt.AddRange(((Stl2)x).dt);
	public void Set(string[] t)=>Stl2.t=t;
	public bool exist{get{return dt.Count!=0;}}
	public int cnt{get{return dt.Count;}}
	public override string ToString(){return string.Join(" ",dt);}
}
public class Stl3:Stl{
	static private string[] t;
	public List<int> dt=new List<int>();
	public void Add(int x)=>dt.Add(x);
	public void AddRange(Stl x)=>dt.AddRange(((Stl3)x).dt);
	public void Set(string[] t)=>Stl3.t=t;
	public bool exist{get{return dt.Count!=0;}}
	public int cnt{get{return dt.Count;}}
	public override string ToString(){return string.Join(" ",dt);}
}
public class Stl4:Stl{
	static private string[] t;
	public bool dt;
	public void Add(int x)=>dt=true;
	public void AddRange(Stl x)=>dt|=((Stl4)x).dt;
	public void Set(string[] t)=>Stl4.t=t;
	public bool exist{get{return dt;}}
	public int cnt{get{return dt?1:0;}}
	public override string ToString(){return dt?"*":"";}
}
public class Pt{
	private StringBuilder sb=new StringBuilder();
	public Pt(Action<Pt> f){f(this);}
	public void W(int s){sb.Append(s);}
	public void W(long s){sb.Append(s);}
	public void W(double s){sb.Append(s);}
	public void W(decimal s){sb.Append(s);}
	public void W(char s){sb.Append(s);}
	public void W(string s){sb.Append(s);}
	public void W(object s){sb.Append(s);}
	public void W(params object[] s){sb.Append(string.Join(" ",s));}
	public void Wl(int s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(long s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(double s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(decimal s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(char s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(string s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(object s){sb.Append(s).Append(Environment.NewLine);}
	public void Wl(int[] s){sb.Append(string.Join(" ",s)).Append(Environment.NewLine);}
	public void Wl(long[] s){sb.Append(string.Join(" ",s)).Append(Environment.NewLine);}
	public void Wl(double[] s){sb.Append(string.Join(" ",s)).Append(Environment.NewLine);}
	public void Wl(string[] s){sb.Append(string.Join(" ",s)).Append(Environment.NewLine);}
	public void Wl(params object[] s){sb.Append(string.Join(" ",s)).Append(Environment.NewLine);}
	public void Wl(){sb.Append(Environment.NewLine);}
	public void Sm(){Console.Write(sb);}
	public void Op(StreamWriter sw){
		sw.Write(sb);
		sw.Close();
	}
	public static bool operator==(Pt a,Pt b){return a.sb.ToString()==b.sb.ToString();}
	public static bool operator!=(Pt a,Pt b){return a.sb.ToString()!=b.sb.ToString();}
	public override bool Equals(object obj){return false;}
	public override int GetHashCode(){return 0;}
}
public class Sc{
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	protected virtual string Rl(){return Console.ReadLine();}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	protected virtual string[] Sp(string st){return st.Split();}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	private T Ct<T>(string s){return (T)Convert.ChangeType(s,typeof(T));}
	public virtual int I{get{return int.Parse(Rl());}}
	public virtual long L{get{return long.Parse(Rl());}}
	public virtual double D{get{return double.Parse(Rl());}}
	public virtual string S{get{return Rl();}}
	public int[] Ia{get{return Array.ConvertAll(Sp(Rl()),int.Parse);}}
	public long[] La{get{return Array.ConvertAll(Sp(Rl()),long.Parse);}}
	public double[] Da{get{return Array.ConvertAll(Sp(Rl()),double.Parse);}}
	public string[] Sa{get{return Sp(Rl());}}
	public object[] Oa{get{return Sp(Rl());}}
	public int[] Ia2{get{return Array.ConvertAll(Sp("0 "+Rl()+" 0"),int.Parse);}}
	public int[] Ia3(string a,string b){return Array.ConvertAll(Sp(a+Rl()+b),int.Parse);}
	public int[] Ia3(int a){return Array.ConvertAll(Sp(Rl()+" "+a.ToString()),int.Parse);}
	public long[] La2{get{return Array.ConvertAll(Sp("0 "+Rl()+" 0"),long.Parse);}}
	public long[] La3(string a,string b){return Array.ConvertAll(Sp(a+Rl()+b),long.Parse);}
	public long[] La3(int a){return Array.ConvertAll(Sp(Rl()+" "+a.ToString()),long.Parse);}
	public double[] Da2{get{return Array.ConvertAll(Sp("0 "+Rl()+" 0"),double.Parse);}}
	public double[] Da3(string a,string b){return Array.ConvertAll(Sp(a+Rl()+b),double.Parse);}
	public T[] Arr<T>(int n,Func<T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f();}return a;}
	public T[] Arr<T>(int n,Func<int,T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(i);}return a;}
	public T[] Arr<T>(int n,Func<string[],T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(Sp(Rl()));}return a;}
	public T[] Arr<T>(int n,Func<int,string[],T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(i,Sp(Rl()));}return a;}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T,T) Tp2<T>(){var s=Sp(Rl());return (Ct<T>(s[0]),Ct<T>(s[1]));}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T,T,T) Tp3<T>(){var s=Sp(Rl());return (Ct<T>(s[0]),Ct<T>(s[1]),Ct<T>(s[2]));}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T,T,T,T) Tp4<T>(){var s=Sp(Rl());return (Ct<T>(s[0]),Ct<T>(s[1]),Ct<T>(s[2]),Ct<T>(s[3]));}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T,T,T,T,T) Tp5<T>(){var s=Sp(Rl());return (Ct<T>(s[0]),Ct<T>(s[1]),Ct<T>(s[2]),Ct<T>(s[3]),Ct<T>(s[4]));}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T,T,T,T,T,T) Tp6<T>(){var s=Sp(Rl());return (Ct<T>(s[0]),Ct<T>(s[1]),Ct<T>(s[2]),Ct<T>(s[3]),Ct<T>(s[4]),Ct<T>(s[5]));}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T1,T2) Tp2<T1,T2>(){var s=Sp(Rl());return (Ct<T1>(s[0]),Ct<T2>(s[1]));}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	public (T1,T1,T2) Tp3<T1,T2>(){var s=Sp(Rl());return (Ct<T1>(s[0]),Ct<T1>(s[1]),Ct<T2>(s[2]));}
}
public class Scr:Sc{
	private List<string> li=new List<string>();
	private int l=0;
	private bool bo=false;
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	protected override string Rl(){
		if(bo){return li[l++%li.Count];}
		li.Add(Console.ReadLine());
		return li[li.Count-1];
	}
	public void Again(){bo=true;}
}
public class Scs:Sc{
	private StreamReader sr;
	public Scs(string t){sr=new StreamReader(t);}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	protected override string Rl(){return sr.ReadLine();}
	public void Close(){sr.Close();}
}
public class Sc2:Sc{
	private string[] sps=new string[]{" ","　","\t"};
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	protected override string[] Sp(string st){return st.Split(sps,StringSplitOptions.RemoveEmptyEntries);}
	public override int I{get{return int.Parse(Sp(Rl())[0]);}}
	public override long L{get{return long.Parse(Sp(Rl())[0]);}}
	public override double D{get{return double.Parse(Sp(Rl())[0]);}}
	public override string S{get{return Sp(Rl())[0];}}
}
public class Scs2:Sc2{
	private StreamReader sr;
	public Scs2(string t){sr=new StreamReader(t);}
	[MethodImpl(MethodImplOptions.AggressiveInlining)]
	protected override string Rl(){return sr.ReadLine();}
	public void Close(){sr.Close();}
}
public class Sct:Sc{
	private List<string> li=new List<string>();
	private int l=0;
	public void Add(int s){li.Add(s.ToString());}
	public void Add(long s){li.Add(s.ToString());}
	public void Add(double s){li.Add(s.ToString());}
	public void Add(string s){li.Add(s.ToString());}
	public void Add(object s){li.Add(s.ToString());}
	public void Add(int[] s){li.Add(string.Join(" ",s));}
	public void Add(long[] s){li.Add(string.Join(" ",s));}
	public void Add(double[] s){li.Add(string.Join(" ",s));}
	public void Add(string[] s){li.Add(string.Join(" ",s));}
	public void Add(params object[] s){li.Add(string.Join(" ",s));}
	protected override string Rl(){return li[l++];}
	public void Clear(){li.Clear();l=0;}
	public void Again(){l=0;}
	public void Pf(){Console.WriteLine(string.Join("\n",li));}
}
0