平时一直用Scanner类输入输出,但在输入输出数据量很大的情况下,就会导致Time Limit Exceeded(超时)。
以一个并查集题目为例。
题目描述
或许你并不知道,你的某个朋友是你的亲戚。
他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。
如果能得到完整的家谱,判断两个人是否是亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及。
在这种情况下,最好的帮手就是计算机。
为了将问题简化,你将得到一些亲戚关系的信息,如Marry和Tom是亲戚,Tom和Ben是亲戚,等等。
从这些信息中,你可以推出Marry和Ben是亲戚。
请写一个程序,对于我们的关于亲戚关系的提问,以最快的速度给出答案。
输入格式
输入由两部分组成。
第一部分以 N,M 开始。N 为问题涉及的人的个数。这些人的编号为 1,2,3,…,N。下面有 M 行,每行有两个数 ai,bi,表示已知 ai
和 bi 是亲戚。
第二部分以 Q 开始。以下 Q 行有 Q 个询问,每行为 ci,di,表示询问 ci 和 di 是否为亲戚。
输出格式
对于每个询问 ci,di,输出一行:若 ci 和 di 为亲戚,则输出“Yes”,否则输出“No”。
数据范围
1≤N≤20000,
1≤M≤1e6,
1≤Q≤1e6
解读
简单的并查集问题,但输入输出的量太大,用Scanner
会超时。
解决方案
可选用两种快读方式,一种是快读一个数字,另一种是快读一行。建议选用快读一行,更快些。
快读一行(建议使用)
import java.io.*;
import java.util.Scanner;
public class Main {
public static int N = 20010;
public static int[] p = new int[N];
public static void main(String[] args) throws IOException {
BufferedReader bw = new BufferedReader(new InputStreamReader(System.in));
PrintWriter pw = new PrintWriter(new OutputStreamWriter(System.out));
String[] c = bw.readLine().split(" ");
int n = Integer.parseInt(c[0]);
int m = Integer.parseInt(c[1]);
for (int i = 1; i <= n; i ++)
p[i] = i;
while (m -- > 0){
String[] s = bw.readLine().split(" ");
int a = Integer.parseInt(s[0]);
int b = Integer.parseInt(s[1]);
// 让b祖先当 a祖先的祖先
p[find(a)] = find(b);
}
int q = Integer.parseInt(bw.readLine());
while (q -- > 0){
String[] s = bw.readLine().split(" ");
int a = Integer.parseInt(s[0]);
int b = Integer.parseInt(s[1]);
if (find(a) == find(b))
pw.println("Yes");
else
pw.println("No");
}
pw.flush();
}
/**
* 路径压缩
* @param x
* @return
*/
private static int find(int x) {
if (p[x] != x)
p[x] = find(p[x]);
return p[x];
}
}
快读每个数字
import java.io.*;
import java.util.Scanner;
public class Main {
public static int N = 20010;
public static int[] p = new int[N];
public static void main(String[] args) throws IOException {
StreamTokenizer st = new StreamTokenizer(new InputStreamReader(System.in));
PrintWriter pw = new PrintWriter(new OutputStreamWriter(System.out));
st.nextToken(); int n = (int) st.nval;
for (int i = 1; i <= n; i ++)
p[i] = i;
st.nextToken(); int m = (int) st.nval;
while (m -- > 0){
st.nextToken(); int a = (int) st.nval;
st.nextToken(); int b = (int) st.nval;
// 让b祖先当 a祖先的祖先
p[find(a)] = find(b);
}
st.nextToken(); int q = (int) st.nval;
while (q -- > 0){
st.nextToken(); int a = (int) st.nval;
st.nextToken(); int b = (int) st.nval;
if (find(a) == find(b))
pw.println("Yes");
else
pw.println("No");
}
pw.flush();
}
/**
* 路径压缩
* @param x
* @return
*/
private static int find(int x) {
if (p[x] != x)
p[x] = find(p[x]);
return p[x];
}
}
Attention
- StreamTokenizer 类中 有个方法是 nval 这个方法就是我们读取数据的方法,他默认是double类型的,所以我们在读入的时候要将他强制转换类型。
- nextToken()方法,这个方法是我们每次读入数据之前必须要写的,也就是我们读入一个数据之前就要写一个这个方法才行。
- 快输出PrintWriter输出时 记得最后加个flush()。
- 也可以是用BufferedReader readLine后,再split。建议使用这个。少写语句,且读一行更快。