Java實現藍橋杯G將軍的示例程式碼
阿新 • • 發佈:2020-02-12
G將軍有一支訓練有素的軍隊,這個軍隊除開G將軍外,每名士兵都有一個直接上級(可能是其他士兵,也可能是G將軍)。現在G將軍將接受一個特別的任務,需要派遣一部分士兵(至少一個)組成一個敢死隊,為了增加隊員的獨立性,要求如果一名士兵在隊中,他的直接上級不能在隊中。
請問,G將軍有多少種派出隊的方法。注意,G將軍也可以作為一個士兵進入隊。
輸入格式
輸入的第一行包含一個整數n,表示包括G將軍在內的軍隊的人數。軍隊的士兵從1至n編號,G將軍編號為1。
接下來n-1個數,分別表示編號為2,3,…,n的士兵的直接上級編號,編號i的士兵的直接上級的編號小於i。
輸出格式
輸出一個整數,表示派出隊的方案數。由於數目可能很大,你只需要輸出這個數除10007的餘數即可。
3
1 1
樣例輸出1
4
樣例說明
這四種方式分別是:
選1;
選2;
選3;
選2,3。
樣例輸入2
7
1 1 2 2 3 3
樣例輸出2
40
資料規模與約定
對於20%的資料,n ≤ 20;
對於40%的資料,n ≤ 100;
對於100%的資料,1 ≤ n ≤ 100000。
資源約定:
峰值記憶體消耗(含虛擬機器) < 256M
CPU消耗 < 2000ms
請嚴格按要求輸出,不要畫蛇添足地列印類似:“請您輸入…” 的多餘內容。
所有程式碼放在同一個原始檔中,除錯通過後,拷貝提交該原始碼。
注意:不要使用package語句。不要使用jdk1.7及以上版本的特性。
注意:主類的名字必須是:Main,否則按無效程式碼處理。
import java.util.ArrayList; import java.util.Scanner; public class Main { public static int n; public static int MOD = 10007; public static ArrayList<Integer>[] list; public static long[][] dp; public void dfs(int root) { dp[root][0] = 1; dp[root][1] = 1; for(int i = 0;i < list[root].size();i++) { int child = list[root].get(i); dfs(child); dp[root][0] = dp[root][0] * (dp[child][0] + dp[child][1]) % MOD; dp[root][1] = dp[root][1] * dp[child][0] % MOD; } } @SuppressWarnings("unchecked") public static void main(String[] args) { Main test = new Main(); Scanner in = new Scanner(System.in); n = in.nextInt(); list = new ArrayList[n + 1]; for(int i = 1;i <= n;i++) list[i] = new ArrayList<Integer>(); for(int i = 2;i <= n;i++) { int father = in.nextInt(); list[father].add(i); } dp = new long[n + 1][2]; test.dfs(1); long result = (dp[1][0] + dp[1][1] - 1) % MOD; System.out.println(result); } }
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。