問題
次元の行ベクトル と正の整数 が与えられます。 行列 の要素のうち、 であるものの個数を求めてください。
は の転置 を表します。
制約
- 入力は全て整数
入力
入力は標準入力から以下の形式で与えられます。
出力
問題の答えを標準出力に出力してください。
入出力例
例 1
入力
3 1 0 1 2 -4 0 0
出力
7
より、求める の個数は です。
例 2
入力
3 10 0 1 2 -4 0 0
出力
9
を 乗した行列は全ての要素が となるため、求める の個数は です。
例 3
入力
1 4 2 -2
出力
0
例 4
入力
32 19 -13699 -3979 0 -36322 -41682 -59976 -95756 18850 65543 36711 77362 -20053 26682 -50531 0 -81945 46356 -85638 -52312 25462 47893 -81086 88224 -61133 -20964 55553 -69852 -84006 -60646 -23686 -87149 19487 -91594 -48022 -49368 -80022 -71533 11696 11127 11687 -10397 -79284 -90586 0 -31655 -10128 93468 -22952 -17927 -32621 -21006 73914 -56260 -258 -15423 36802 0 57619 -80453 81613 32278 61678 0 326
出力
154
提出
登録なしで提出できます。
Go (1.14)