不是哥们，这对吗

#include<bits/stdc++.h>
using namespace std;
#define int long long

signed main(){

    int t;
    cin >> t;
    while(t--){
        int n;
        string s;
        cin >> n >> s;
        int x = 0;
        int y = 0;
        for (int i = 0;i < n;i++){
            if (s[i] == '0'){
                x ++;
            } else {
                y++;
            }
        }
        int mx = INT_MIN;
        if (x > 0 && y > 0){
            mx = max(mx, x * y);
        }
        if (x > 0 && y == 0){
            mx = max(mx, x * x);
        }
        if (x == 0 && y > 0){
            mx = max(mx, y * y);
        }
        cout << mx << endl;
    }
    
    return 0;
}

应该输出：
9
12
16
12
9
1
实际输出：
6
12
8
12
3
1