最近在做一道最近公共祖先算法题的时候,遇到了StackOverFlow。
附上题目和代码:
import java.io.*;
import java.util.*;
public class Main {
static int maxn=100001;
static int n,m,cnt;
static int[] cow=new int[maxn]; //0更赛牛G 第一位 1 H 二进制第二位
static int[] vv=new int[maxn<<1],to=new int[maxn<<1];
static int[] he=new int[maxn],lg=new int[maxn];
static int[] depth=new int[maxn];
static int[][] fa = new int[maxn][20],dis=new int[maxn][20];
public static void main(String[] args) throws IOException {
n=nextInt();m=nextInt();
String line = nextString();
for(int i=1;i<=n;i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i?1:0);
for(int i=0;i<line.length();i++)
cow[i+1]=line.charAt(i);
for(int i=1;i<n;i++){
int u=nextInt(),v=nextInt();
addEdge(u,v);
addEdge(v,u);
}
dfs(1,0);
int x,y,z;
for(int i=1;i<=m;i++){
x=nextInt();y=nextInt();z=nextString().charAt(0);
int ans = lca(x, y);
if(z=='G') ans=ans&1;else ans=ans&2;
if(ans>0) out.print(1);
else out.print(0);
}
out.flush();
}
static void dfs(int u,int fath){
fa[u][0]=fath;depth[u]=depth[fath]+1;
if(cow[u]=='G') dis[u][0]=dis[u][0]|1; else dis[u][0]=dis[u][0]|2;
if(cow[fath]=='G') dis[u][0]=dis[u][0]|1; else dis[u][0]=dis[u][0]|2;
for(int i=1;i<=lg[depth[u]];i++){
dis[u][i]=dis[u][i-1]|dis[fa[u][i-1]][i-1];
fa[u][i]=fa[fa[u][i-1]][i-1];
}
for(int i=he[u];i>0;i=to[i])
if(vv[i]!=fath) dfs(vv[i],u);
}
static int lca(int x,int y){
int ans=0;
if(depth[x]<depth[y]){
int t=x;x=y;y=t;
}
while(depth[x]>depth[y]){
ans=ans|dis[x][lg[depth[x]-depth[y]]-1];
x=fa[x][lg[depth[x]-depth[y]]-1];
}
if(x==y) {
if(cow[x]=='G') ans|=1;
else ans|=2;
return ans;
}
for(int k=lg[depth[x]]-1;k>=0;k--){
if(fa[x][k]!=fa[y][k]){
ans=ans|dis[x][k];
ans=ans|dis[y][k];
x=fa[x][k];
y=fa[y][k];
}
}
ans=ans|dis[x][0];
ans=ans|dis[y][0];
return ans;
}
static void addEdge(int u,int v){
vv[++cnt]=v;
to[cnt]=he[u];
he[u]=cnt;
}
static BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
static StreamTokenizer in = new StreamTokenizer(br);
static PrintWriter out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
public static int nextInt() throws IOException{
in.nextToken();
return (int)in.nval;
}
public static String nextString() throws IOException {
in.nextToken();
return in.sval;
}
}
一开始以为是数据量过大,idea接收不了那么多行数据导致出错,于是我改了idea输入行限制,再然后就看到了报错:java.lang.StackOverFlow。然后我就想明白,噢,原来是递归太深了,栈容量不够用,然后我改了idea最大栈容量-Xss100m,终于得到了正确答案。但是题目依然没办法完全AC:
我可以改自己电脑上idea的-xss,但是代码是提交到出题方那里的...所以用倍增最近公共祖先算法是做不出这题目的,除非是c++。当然,可以用并查集和连通块的思想去建模求解,但是蒟蒻暂时不会连通块。选择一个好的思路去建模真的很重要!!!!!
后面我还是换c++写把这题解决了,不过我对java的理解更深了一些,为此查阅了很多很多资料,不仅对输入流有了新的认识,还对虚拟机堆栈理解更进一步,特地记录下来。