ACM-ICPC 2018青岛网络赛-H题

#include <iostream>
#include <string>
#include <string.h>

using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    int t;
    cin >> t;
    while (t--)
    {
        int a, b;
        string str;
        cin >> str;
        int num = 0;
        long long sum = 0;
        for (int i = 0; i < str.size(); i++)
        {
            a = str[i] - ‘0‘;
            if (!a)
                a = 2;
            if (i)
            {
                if (str[i] == str[i - 1])
                {
                    b = 2;
                }
                else
                {
                    b = 1;
                }
            }
            sum += a * (str.size() - i) + b * (str.size() - i) * num;
            num++;
        }
        cout << sum << endl;
    }
    return 0;
}