結果

問題 No.493 とても長い数列と文字列(Long Long Sequence and a String)
ユーザー parukiparuki
提出日時 2017-03-12 20:33:10
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 800 ms
コード長 2,743 bytes
コンパイル時間 1,509 ms
コンパイル使用メモリ 169,828 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-30 00:34:36
合計ジャッジ時間 3,971 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 115
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"
using namespace std;
#define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i))
#define rep(i,j) FOR(i,0,j)
#define each(x,y) for(auto &(x):(y))
#define mp make_pair
#define mt make_tuple
#define all(x) (x).begin(),(x).end()
#define debug(x) cout<<#x<<": "<<(x)<<endl
#define smax(x,y) (x)=max((x),(y))
#define smin(x,y) (x)=min((x),(y))
#define MEM(x,y) memset((x),(y),sizeof (x))
#define sz(x) (int)(x).size()
#define pb push_back
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<ll> vll;

const int MOD = (int)1e9 + 7;
int add(int a, int b){
    int c = a + b;
    if(c >= MOD)c -= MOD;
    else if(c < 0)c += MOD;
    return c;
}
void sadd(int &a, int b){
    a += b;
    if(a >= MOD)a -= MOD;
    else if(a < 0)a += MOD;
}
int mul(int a, int b){
    return (int)((long long)a*b%MOD);
}
void smul(int &a, int b){
    a = (int)((long long)a*b % MOD);
}
int powm(int a, long long k){
    long long res=1, b = a;
    while(k){
        if(k&1)res=res*b%MOD;
        b=b*b%MOD;
        k >>= 1;
    }
    return (int)res;
}
int inverse(int x){
    long long a = x, b = MOD, u = 1, v = 0;
    while(b) {
        long long t = a / b;
        a -= t * b;
        swap(a, b);
        u -= t * v;
        swap(u, v);
    }
    if(u < 0)u += MOD;
    return (int)u;
}

ll K, L, R;
ll cnt[61][10];
ll len[61];

pair<ll, int> f(ll r) {
    pair<ll, int> res(0, 1);

    if(r == 0) res;

    int k = -1;
    while(len[k+1] <= r) {
        k++;
    }

    r -= len[k];
    if(k + (r>0?1:0) > K) {
        cout << -1 << endl;
        exit(0);
    }

    rep(i, 10) {
        int x = i == 0 ? 10 : i;
        res.first += cnt[k][i] * x;
        smul(res.second, powm(x, cnt[k][i]));
    }

    if(r > 0) {
        int val = (k + 1)*(k + 1);
        string s = to_string(val);
        for(int i = 0; i < sz(s) && r>0; i++) {
            r--;
            int x = s[i] == '0' ? 10 : s[i] - '0';
            res.first += x;
            smul(res.second, x);
        }
    }

    if(r > 0) {
        auto sub = f(r);
        res.first += sub.first;
        smul(res.second, sub.second);
    }
    return res;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> K >> L >> R;

    len[1] = 1;
    cnt[1][1] = 1;
    for(int i = 2; i <= 60; ++i) {
        int val = i*i;
        len[i] = len[i - 1] * 2 + sz(to_string(i*i));
        rep(j, 10) {
            cnt[i][j] = cnt[i - 1][j] * 2;
        }
        while(val > 0) {
            cnt[i][val % 10]++;
            val /= 10;
        }
    }

    auto res_r = f(R), res_l = f(L - 1);

    ll sum = res_r.first - res_l.first;
    int mulmul = mul(res_r.second, inverse(res_l.second));
    cout << sum << ' ' << mulmul << endl;
}
0