在一個小農村裡,農夫們喜歡栽種彩色的花朵,其中每個農夫會將他的農場裡都種滿同一種顏色的花朵。由於地區發展問題,因此並不是每個農場都是可以直接相通的,有時候需要先經過其他農場繞路,或是有時候甚至沒有道路可以到達。今天市長來訪查,一共會走過三個農場 $u, v, w$,農村裡的人們希望讓市長可以走過花朵顏色都不相同,並且相互相鄰連成一條連續路徑的三個農場(也就是說,$c_u, c_v, c_w$ 兩兩相異,且 $(u, v)$ 跟 $(v, w)$ 之間都有直接相連的道路 ),並把這個探訪路徑 $u \to v \to w$ 稱為彩色路徑,因此他們想要統計農村裡共有多少個彩色路徑,但是彩色路徑數量可能會非常多,因此他們請求你幫他們計算。
由於發展經費問題,因此農村裡不會蓋設無意義的道路,也就是不會有道路兩端連接一樣的農場,也不會有兩條不同道路連接一樣的起點與終點。
請注意 $a \to b \to c$ 和 $c \to b \to a$ 會視作兩條不同路徑。
第一行會輸入兩個正整數 $N$ 和 $M$,代表農場的數量還有道路的數量。
第二行會輸入 $N$ 個正整數 $c_i$,代表每個農場的花朵顏色。
接下來會輸入 $M$ 行,每行有兩個數字 $a_i, b_i$,代表一條雙向的道路連接的兩個農場。
輸出一個數字代表總共有幾行彩虹路徑。
保證路徑的數量不會超過 $10^ {18}$。
3 2 1 2 3 1 2 2 3
2
5 4 1 1 1 1 1 1 2 2 3 3 4 4 5
0
5 7 1 2 3 4 1 1 2 2 3 3 4 4 5 1 3 3 5 2 5
24
範例測資一中:
1-2-3 以及 3-2-1 為兩條彩虹路徑,因此輸出 2。
範例測資二中:
由於全部的顏色都是 1,因此沒有彩虹路徑,輸出 0。
YTP 2025 國中組初賽 p3
No. | Testdata Range | Constraints | Score |
---|---|---|---|
1 | 0~2 | 範例測試資料 | 0 |
2 | 0~41 | $N \leq 2000, M \leq 2000$ | 6 |
3 | 0~110 | 無額外限制 | 9 |