要解决Android Studio中Java代码显示为红色并显示需要设置JDK的问题，可以按照以下步骤操作：

1. 首先，确保已经安装了Java Development Kit（JDK）并配置了环境变量。可以从Oracle的官方网站上下载并安装适合您操作系统的JDK版本。

2. 打开Android Studio，并在顶部菜单栏中选择“File”（文件）选项，然后选择“Project Structure”（项目结构）。

3. 在弹出的对话框中，选择“SDK Location”（SDK位置）选项卡。

4. 在“JDK Location”（JDK位置）一栏中，点击“...”按钮。

5. 浏览并选择您安装的JDK的路径。

6. 点击“Apply”（应用）按钮来保存更改。

7. 关闭并重新打开Android Studio。

此时，Java代码应该不再显示为红色，而是正常显示了。

以下是一个简单的Java代码示例：

public class MainActivity extends AppCompatActivity {
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        
        // 在这里编写您的Java代码
        int a = 10;
        int b = 20;
        int sum = a + b;
        System.out.println("两个数的和为：" + sum);
    }
}

请注意，上述步骤和代码示例仅适用于解决Android Studio中Java代码显示为红色并显示需要设置JDK的问题。如果问题仍然存在或有其他错误，请检查您的JDK安装和配置是否正确，并确保Android Studio的设置正确。