에디트 텍스트 값에따른 이미지 변환

Question

메인 을 이렇게해서 에디트 텍스트에 값 A 또는 B  입력시 서브 액티비티에 이미지를 다르게 출력하려고 합니다 저기 함수를 어떤걸 넣어야될지모르겟어요 뭐 getImage 같은것도없고 이것저것해봣는데 도저히 안되서 질문드립니다 

btn = (Button)findViewById(R.id.button);
tx = (TextView)findViewById(R.id.EditText);

btn.setOnClickListener(new View.OnClickListener() {
    @Override
    public void onClick(View v) {
        Intent i = new Intent(MainActivity.this , subActivity.class);
        i.putExtra("text",tx.getText().toString() );
        startActivity(i);
    }
});

subActivity.java

private ImageView img;
private Button btn;
int Images = R.drawable.cat;
int Images2 = R.drawable.cat2;
@Override
protected void onCreate(Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);
    setContentView(R.layout.activity_main);
    btn = (Button)findViewById(R.id.button);
    img = (ImageView)findViewById(R.id.imageView);


    Intent i = getIntent();
    String txA = i.getStringExtra("text");

    if(i.toString().equals("A")){
        
    }else if(i.toString().equals("B")){

    }

    btn.setOnClickListener(new View.OnClickListener() {
        @Override
        public void onClick(View v) {
            Intent intent = new Intent(subActivity.this , MainActivity.class);
            startActivity(intent);
            finish();
        }
    });
}

Answer 1

아래 코드에 Strign check 하나 만든후에

if 문에 A일경우 1, B일경우 2 줘서 그 값을 통해 버튼에서 작업하면 안되나요?