We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
Construct the Array
Construct the Array
Sort by
recency
|
168 Discussions
|
Please Login in order to post a comment
how to avoid recursion:
how to avoid recusrion?
!/bin/python3
import math import os import random import re import sys
#
Complete the 'countArray' function below.
#
The function is expected to return a LONG_INTEGER.
The function accepts following parameters:
1. INTEGER n
2. INTEGER k
3. INTEGER x
#
def _build_dp(last, s, array, rest, x, n): if len(rest)>n: return 0 if n==2: if s!=x: return 1 else: return 0
_count=0 if (len(rest)==n): for i in rest: if i!=s: _count+=_build_dp(s, i, array, [ j for j in rest if j!=i ], x, n-1) else: for i in array: if i!=s: _count+=_build_dp(s, i, array, [ j for j in rest if j!=i ], x, n-1) return _count
def countArray(n, k, x): _count=0 for i in range(2, k+1): _count+=_build_dp(1, i, range(1, k+1), [ i for i in range(2, k+1) if i!=x], x, n-1) return _count
if name == 'main': fptr = open(os.environ['OUTPUT_PATH'], 'w')
public static long countArray(int n, int k, int x) { long mode = 1000000000 + 7; long[][] dp = new long[n + 1][2]; if (n == 2) { if (x == 1) return 0; else return 1; } if (x != 1) { dp[2][0] = 1; dp[2][1] = k - 2; } else { dp[2][0] = 0; dp[2][1] = k - 1; } for (int i = 3; i <= n; i++) { dp[i][0] = dp[i - 1][1] % mode; dp[i][1] = (dp[i - 1][1] * (k - 2) + dp[i - 1][0] * (k - 1)) % mode; }
I don't know where this code is wrong. I modularized it, but the answer is different.