QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#596993#9225. Fibonacci Fusionrikka_lyly#WA 85ms87184kbC++201.9kb2024-09-28 16:50:172024-09-28 16:50:20

Judging History

你现在查看的是最新测评结果

  • [2024-09-28 16:50:20]
  • 评测
  • 测评结果:WA
  • 用时:85ms
  • 内存:87184kb
  • [2024-09-28 16:50:17]
  • 提交

answer

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

#define ll long long
#define ull unsigned long long
// const ll MOD = 4179340454199820289;
const ll MOD = 998244353;
#define __int128 ll
#define MAXN 10000010
#define C 20

ll fib[MAXN];
void pre()
{
    fib[0] = 0, fib[1] = 1;
    for (int i = 2; i < MAXN; i++)
    {
        fib[i] = ((__int128)fib[i - 2] + fib[i - 1]) % MOD;
    }
}

__int128 trans(const string& s)
{
    __int128 ans = 0;
    for (int i = 0; i < s.length(); i++)
    {
        (ans *= 10) %= MOD;
        (ans += s[i] - '0') %= MOD;
    }
    return ans;
}

const double lgB = log10((1 + sqrt(5)) / 2);

void solve()
{
    int n;
    cin >> n;
    map<ll, int> sts;
    vector<string> ss(n);
    for (int i = 0; i < n; i++)
    {
        cin >> ss[i];
    }
    vector<int> order(n);
    iota(order.begin(), order.end(), 0);
    sort(order.begin(), order.end(), [&](int i, int j)->bool
         {
        if(ss[i].length() != ss[j].length())
            return ss[i].length() < ss[j].length();
        // if(ss[i].length() >= 1 && ss[i][0] != ss[j][0])
        //     return ss[i][0] < ss[j][0];
        // if(ss[i].length() >= 2 && ss[i][1] != ss[j][1])
        //     return ss[i][1] < ss[j][1];
        // return true;
        return ss[i] < ss[j];
         });
    ll ans = 0;
    for (int i : order)
    {
        const string &s = ss[i];
        ll x = trans(s);
        int p = s.length() / lgB;
        for (int j = max(0, p - C); j < min(p + C, MAXN); j++)
        {
            ll y = (fib[j] - x + MOD) % MOD;
            if(sts.contains(y))
                ans += sts[y];
        }
        sts[x]++;
    }
    cout << ans << '\n';
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    pre();
    int t = 1;
    while (t--)
    {
        solve();
    }
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 30ms
memory: 81716kb

input:

6
50
8
8
5
72
354224848179261915070

output:

4

result:

ok 1 number(s): "4"

Test #2:

score: 0
Accepted
time: 65ms
memory: 86576kb

input:

28
200878223506436882933619847964496455022155117513398820563747455993172799881403389571477889821109288771413214004090719097929400406252135763028179112130390003528046316900603668569910008417315162907579003880220844686222148696041857432602133894827753998572080650383305777912447151917272483538029469449...

output:

27

result:

ok 1 number(s): "27"

Test #3:

score: -100
Wrong Answer
time: 85ms
memory: 87184kb

input:

5187
2640352926124261912741724778991366987330659389621881876017670644497364093930668042530271338851702874394631009332660937266680740235862107353443518003194307853104942996827176097428402408674756368623972812842571069642405111849826172879369776309763468485788964245903781380419155348915131587410703749...

output:

6077

result:

wrong answer 1st numbers differ - expected: '6073', found: '6077'