题解 | 小红的密码修改

#include <cctype>
#include <iostream>
#include <string>
using namespace std;
void solve() {
    string s; cin >> s;
    int n = s.size();
    s = ' ' + s;
    int upper = 0, lower = 0, punct = 0, digit = 0;
    for(int i=1;i<=n;i++) {
        if(isupper(s[i])) upper++;
        else if(islower(s[i])) lower++;
        else if(ispunct(s[i])) punct++;
        else digit++;
    }
    long long ans = 0;
    for(int i=1;i<=n;i++) {
        if(isupper(s[i]) && upper == 1) {
            ans += 25;
        } else if(islower(s[i]) && lower == 1) {
            ans += 25;
        } else if(isdigit(s[i]) && digit == 1) {
            ans += 9;
        } else if(ispunct(s[i]) && punct == 1) {
            ans += 3;
        } else {
            ans += 26 + 26 + 10 + 4 - 1;
        }
    }
    cout << ans << endl;
}
int main() {
    int t; cin >> t;
    while(t--) solve();
}