제출 #299113

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
299113		AaronNaidu	회문 (APIO14_palindrome)	C++14	0 / 100	1089 ms	3228 KiB

palindrome

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

ll mod1 = 1000000007;
ll mod2 = 1000000009;
string s;
ll n;
ll prefSums[20000], sufSums[20000];
ll prefSums2[20000], sufSums2[20000];
ll pows[20000], pows2[20000];
unordered_map<ll, ll> m;

ll divide2(ll num, ll den) {
    ll inv = 1;
    ll exp = den;
    for (int i = 0; i < 32; i++)
    {
        if (mod2-2 & 1<<i)
        {
            inv *= exp;
            inv %= mod2;
        }
        exp *= exp;
        exp %= mod2;
    }
    ll ans = num * exp;
    ans %= mod2;
    return ans;
}

ll divide1(ll num, ll den) {
    ll inv = 1;
    ll exp = den;
    for (int i = 0; i < 32; i++)
    {
        if (mod1-2 & 1<<i)
        {
            inv *= exp;
            inv %= mod1;
        }
        exp *= exp;
        exp %= mod1;
    }
    ll ans = num * exp;
    ans %= mod1;
    return ans;
}


int main() {
    cin >> s;
    ll thisPow = 1;
    ll thisPow2 = 1;
    n = s.length();
    for (int i = 0; i < n; i++)
    {
        pows[i]= thisPow;
        thisPow *= 11;
        thisPow %= mod1;
        pows2[i]= thisPow2;
        thisPow2 *= 13;
        thisPow2 %= mod2;
    }
    prefSums[0] = s[0] - 'a' + 1;
    prefSums2[0] = s[0] - 'a' + 1;
    for (int i = 1; i < n; i++)
    {
        prefSums[i] = prefSums[i-1] + (s[i] - 'a' + 1) * pows[i];  
        prefSums[i] %= mod1;  
        prefSums2[i] = prefSums2[i-1] + (s[i] - 'a' + 1) * pows2[i];  
        prefSums2[i] %= mod2; 
    }
    sufSums[n-1] = s[n-1] - 'a' + 1;
    sufSums2[n-1] = s[n-1] - 'a' + 1;
    for (int i = n-2; i >= 0; i--)
    {
        sufSums[i] = sufSums[i+1] + (s[i] - 'a' + 1) * pows[n-1-i];
        sufSums[i] %= mod1;
        sufSums2[i] = sufSums2[i+1] + (s[i] - 'a' + 1) * pows2[n-1-i];
        sufSums2[i] %= mod2;
    }
    /*for (int j = 0; j < n; j++)
    {
        cout << prefSums[j] << " ";
        //cout << prefSums[j] << " " << sufSums[j] << "\n";
    }
    cout << "\n";
    for (int i = 1; i < n; i++)
    {
        for (int j = i; j < n; j++)
        {
            cout << (prefSums[j] - prefSums[i-1]) / pows[i] << " ";
        }
        cout << "\n";
    }
    cout << "\n";
    for (int i = 0; i < n; i++)
    {
        for (int j = i; j < n-1; j++)
        {
            cout << (sufSums[i] - sufSums[j+1]) / pows[n-1-j] << " ";
        }
        cout << sufSums[i];
        cout << "\n";
    }*/
    ll totMax = 0;
    for (int j = 0; j < n-1; j++)
    {
        if (prefSums[j] == divide1(sufSums[0]-sufSums[j+1], pows[n-1-j]) and prefSums2[j] == divide2(sufSums2[0]-sufSums2[j+1], pows2[n-1-j]))
        {
            //cout << "Entered\n";
            ll curr = 0;
            if (m.find(prefSums[j]) != m.end())
            {
                curr = m.find(prefSums[j])->second;
                m.erase(prefSums[j]);
            }
            curr += j+1;
            if (curr > totMax)
            {
                //cout << "New max of " << curr << " from " << 0 << " to " << j << "\n";
            }
            
            totMax = max(totMax, curr);
            m.insert({prefSums[j], curr});
        }
    }
    if (prefSums[n-1] == sufSums[0] and prefSums2[n-1] == sufSums2[0])
    {
        ll curr = 0;
        if (m.find(prefSums[n-1]) != m.end())
        {
            curr = m.find(prefSums[n-1])->second;
            m.erase(prefSums[n-1]);
        }
        curr += n;
        if (curr > totMax)
        {
           // cout << "New max of " << curr << " from " << 0 << " to " << n-1 << "\n";
        }
        totMax = max(totMax, curr);
        m.insert({prefSums[n-1], curr});
    }
    for (int i = 1; i < n; i++)
    {
        for (int j = i; j < n-1; j++)
        {
            if (divide1(prefSums[j]-prefSums[i-1], pows[i]) == divide1(sufSums[i]-sufSums[j+1], pows[n-1-j]) and divide2(prefSums2[j]-prefSums2[i-1], pows2[i]) == divide2(sufSums2[i]-sufSums2[j+1], pows2[n-1-j]))
            {
                ll curr = 0;
                if (m.find((prefSums[j]-prefSums[i-1])/pows[i]) != m.end())
                {
                    curr = m.find((prefSums[j]-prefSums[i-1])/pows[i])->second;
                    m.erase((prefSums[j]-prefSums[i-1])/pows[i]);
                }
                curr += j-i+1;
                if (curr > totMax)
                {
                   // cout << "New max of " << curr << " from " << i << " to " << j << "\n";
                }
                totMax = max(totMax, curr);
                m.insert({(prefSums[j]-prefSums[i-1])/pows[i], curr});
            }
        }
        if (divide1(prefSums[n-1]-prefSums[i-1], pows[i]) == sufSums[i] and divide2(prefSums2[n-1]-prefSums2[i-1], pows2[i]) == sufSums2[i])
        {
            ll curr = 0;
            if (m.find(sufSums[i]) != m.end())
            {
                curr = m.find(sufSums[i])->second;
                m.erase(sufSums[i]);
            }
            curr += n-i;
            if (curr > totMax)
            {
               // cout << "New max of " << curr << " from " << i << " to " << n-1 << "\n";
            }
            totMax = max(totMax, curr);
            m.insert({sufSums[i], curr});
        }
    }
    cout << totMax << "\n";
}

컴파일 시 표준 에러 (stderr) 메시지

palindrome.cpp: In function 'll divide2(ll, ll)':
palindrome.cpp:19:17: warning: suggest parentheses around '-' in operand of '&' [-Wparentheses]
   19 |         if (mod2-2 & 1<<i)
      |             ~~~~^~
palindrome.cpp: In function 'll divide1(ll, ll)':
palindrome.cpp:37:17: warning: suggest parentheses around '-' in operand of '&' [-Wparentheses]
   37 |         if (mod1-2 & 1<<i)
      |             ~~~~^~

• Subtask 1: 0 / 8
• Subtask 2: 0 / 15
• Subtask 3: 0 / 24
• Subtask 4: 0 / 26
• Subtask 5: 0 / 27