Program menghitung nilai x dengan Array dan Function

Buatlah program untuk menghitung nilai x, dimana nilai x merupakan nilai array yang dimasukkan oleh user misalnya sebanyak 5 nilai inputan. Kemudian nilai x tersebut dimasukkan ke dalam fungsi matematika, dimana fungsi matematika tersebut ada 2 yatiu f(x) dan r(x). Adapun rumus kedua fungsi tersebut sebagai berikut:

  1. f(x) = x2 + 2x + 1
  2. r(x) = 3x + 5

Adapun codenya sebagai berikut:

import java.util.Scanner;

/**
 *
 * @author Riki Afriansyah
 */
public class Arrayinfunction {

    /**
     * @param args the command line arguments
     */
    public static void main(String[] args) {
        // TODO code application logic here
        int[] A = new int[5];
        
        Scanner in = new Scanner(System.in); 
        
         // mengisi data ke array
        for( int i = 0; i <A.length; i++ ){
            System.out.print("Input Nilai  ke-" + (i+1) + ": ");
            A[i] = in.nextInt ();
        }
        
        System.out.println("----------------");
        System.out.println("f(x) = x*x + 2*x +1 dan r(x)= 3*x + 5");
           // menampilkan nilai array dengan function f1
        for( int i = 0; i < A.length; i++ ){
            System.out.println("f("+A[i]+") = "+f(A[i])+" dan "+"r("+A[i]+") = "+r(A[i]));
       }
    }
    
    
    public static int f(int x){
        return x*x+2*x+1;
    }
    
    public static int r(int x){
        return 3*x+5;
    }
    
}