PTA 分而治之 Java
分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。
输入格式:
输入在第一行给出两个正整数 N 和 M(均不超过10 000),分别为敌方城市个数(于是默认城市从 1 到 N 编号)和连接两城市的通路条数。随后 M 行,每行给出一条通路所连接的两个城市的编号,其间以一个空格分隔。在城市信息之后给出参谋部的系列方案,即一个正整数 K (≤ 100)和随后的 K 行方案,每行按以下格式给出:
Np v[1] v[2] … v[Np]
其中 Np 是该方案中计划攻下的城市数量,后面的系列 v[i] 是计划攻下的城市编号。
输出格式:
对每一套方案,如果可行就输出YES,否则输出NO。
输入样例:
10 11
8 7
6 8
4 5
8 4
8 1
1 2
1 4
9 8
9 1
1 10
2 4
5
4 10 3 8 4
6 6 1 7 5 4 9
3 1 8 4
2 2 8
7 9 8 7 6 5 4 2
输出样例:
NO
YES
YES
NO
NO
乍一看,并查集,判断连通数是否等于剩余城市数
仔细一想,所有的路都走不通了不就是所有城市都被孤立了吗…
可惜最后一个测试用例超时…
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.util.LinkedList;
import java.util.StringTokenizer;public class Main {
public static void main(String[] args) throws IOException {
Reader.init(System.in);int N = Reader.nextInt();int M = Reader.nextInt();// 读取数据linked_list[] path = new linked_list[N + 1]; // 记录有路通向外面的城市for (int i = 1; i < path.length; i++) {
path[i] = new linked_list();}for (int i = 0; i < M; i++) {
int s = Reader.nextInt();int e = Reader.nextInt();path[s].list.add(e); // 只需要记录一个方向即可path[s].exits = path[e].exits = 1; // 标记是有路的城市}int K = Reader.nextInt();for (int i = 0; i < K; i++) {
// 读取攻打计划int Np = Reader.nextInt();int[] plan = new int[N + 1]; // 记录要攻打的城市for (int j = 0; j < Np; j++) {
plan[Reader.nextInt()] = 1;}boolean isOK = true; // 判断是否可行for (int j = 1; j < path.length && isOK; j++) {
// 如果该城市不是本来就孤立,且不在攻打的计划上if (path[j].exits == 1 && plan[j] != 1) {
// 那么查找是否有城市和该城市相连接for (int item : path[j].list) {
// 如果找到了一个不在攻打计划上的城市和该城市相连,则表示计划不可行if (plan[item] != 1) {
isOK = false;break;}}}}if (isOK) {
System.out.println("YES");} else {
System.out.println("NO");}}}static class linked_list {
int exits = 0;LinkedList<Integer> list = new LinkedList<Integer>();}
}// Class for buffered reading int and double values *//*
class Reader {
static BufferedReader reader;static StringTokenizer tokenizer;// ** call this method to initialize reader for InputStream *//*static void init(InputStream input) {
reader = new BufferedReader(new InputStreamReader(input));tokenizer = new StringTokenizer("");}static void init(File file) {
try {
reader = new BufferedReader(new FileReader(file));} catch (FileNotFoundException e) {
// TODO Auto-generated catch blocke.printStackTrace();}tokenizer = new StringTokenizer("");}// ** get next word *//*static String next() throws IOException {
while (!tokenizer.hasMoreTokens()) {
// TODO add check for eof if necessarytokenizer = new StringTokenizer(reader.readLine());}return tokenizer.nextToken();}static String nextLine() throws IOException {
return reader.readLine();}static int nextInt() throws IOException {
return Integer.parseInt(next());}static char nextChar() throws IOException {
return next().toCharArray()[0];}static float nextFloat() throws IOException {
return Float.parseFloat(next());}static Double nextDouble() throws IOException {
return Double.parseDouble(next());}
}