题解 | 小红的密码修改

#include <iostream>
#include <vector>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    int T;
    cin >> T;
    while (T--) {
        string password;
        cin >> password;
        vector<int> m(4);
        for (char i : password) {
            if (i - '0' < 10 && i - '0' >= 0) {
                m[2]++;
            }
            else if (i - 'a' >= 0 && i - 'a' <= 26) {
                m[1]++;
            }
            else if (i - 'A' >= 0 && i - 'A' <= 26) {
                m[0]++;
            }
            else {
                m[3]++;
            }
        }

        long long ans = 0;
        for (char c : password) {
            if (c >= '0' && c <= '9') {
                if (m[2] >= 2) {
                    ans += 65;
                }
                else {
                    ans += 9;
                }
            }
            else if (c >= 'a' && c <= 'z') {
                if (m[1] >= 2) {
                    ans += 65;
                }
                else {
                    ans += 25;
                }
            }
            else if (c >= 'A' && c <= 'Z') {
                if (m[0] >= 2) {
                    ans += 65;
                }
                else {
                    ans += 25;
                }
            }
            else {
                if (m[3] >= 2) {
                    ans += 65;
                }
                else {
                    ans += 3;
                }
            }
        }
        cout << ans << '\n';
    }
}