L2-1 分而治之(Java)
分数 25
全屏浏览题目切换布局
作者 陈越
单位 浙江大学
分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。
输入格式:
输入在第一行给出两个正整数 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.util.ArrayList; import java.util.Scanner; public class CityConnection { static ArrayList<Integer>[] v; static int[] book; // 判断方案是否可行的函数 static boolean isFeasible(int n) { for (int i = 0; i < n; i++) { for (int j = 0; j < v[i].size(); j++) { // 被遍历到的城市如果没有被标记并且它还有连接的城市,方案就不可行 if (book[i] == 0 && book[v[i].get(j)] == 0) { return false; } } } return true; } public static void main(String[] args) { Scanner scanner = new Scanner(System.in); int n = scanner.nextInt(); // 城市数量 int m = scanner.nextInt(); // 连接关系数量 v = new ArrayList[n]; for (int i = 0; i < n; i++) { v[i] = new ArrayList<>(); } int a, b; // 输入城市连接关系 while (m-- > 0) { a = scanner.nextInt(); b = scanner.nextInt(); v[a].add(b); v[b].add(a); } int k = scanner.nextInt(); // 测试方案数量 int np, city; while (k-- > 0) { np = scanner.nextInt(); // 当前方案中的城市数量 book = new int[n]; // 标记当前方案中的城市 for (int i = 0; i < np; i++) { city = scanner.nextInt(); book[city] = 1; } // 判断方案是否可行并输出结果 if (isFeasible(n)) { System.out.println("YES"); } else { System.out.println("NO"); } } scanner.close(); } }