long long int solve(int N, int K, int *A, int *B);