17西安网络赛 规律题 Maximum Flow

https://nanti.jisuanke.com/t/17118

在第一组中，只有一个2；
第二组中，每2个数出现一个5；
第三组中，每2个数出现一个5，每4个数出现一个17；
第四组中，每2个数出现一个5，每4个数出现一个17，每8个数出现一个65；
第五组中，每2个数出现一个5，每4个数出现一个17，每8个数出现一个65，每16个数出现一个257；
2=2^0+1, 5=2^2+1, 17=2^4+1, 257=2^8+1;

其实我挺快找到规律了，然后疯狂WA三个小时，改了各种地方，终于知道哪错了，，，就是n是10^18，那么其实会累加超过long long范围的数，所以打表的时候应该不停地取模，很蠢，，，

#include<bits/stdc++.h>
using namespace std;
const long long mx=1000000000000000000;
const long long mod=1000000007ll;
map<long long,long long> mp;
void init()
{
    for(long long i=4ll,k=4ll;k<=mx;k<<=1ll,i<<=2ll,i%=mod)
    {
      //  cout<<i+1<<' '<<k+1<<endl;
        mp[i+1]=k+1;
     //   cout<<mp[i+1]<<endl;
    }
  //  cout<<mp[5]<<endl;
}

int main()
{
    init();
    long long n;
    while(~scanf("%lld",&n))
    {
        if(n==2) {
            printf("1\n");continue;
        }
        if(n==3) {
            printf("3\n");continue;
        }
        long long ans=1;
        ans+=2ll*((n>>1)%mod);
        ans%=mod;
      //  cout<<ans<<endl;
        for(long long i=4ll,cnt=2ll;(1ll<<cnt)<=n;i<<=2ll,i%=mod,cnt++)
        {
         //   cout<<endl<<ans<<endl;
        //    cout<<"cnt="<<cnt<<endl;
         //   cout<<"i="<<i<<endl;
            if(n>=mp[i+1]) ans+=(i+1ll)%mod;
            long long num=n>>cnt;
            num<<=1ll;num|=1ll;
            for(long long j=2ll;j<cnt;j++) num<<=1ll;
            num<<=1ll;num|=1ll;
        //   cout<<"num="<<num<<endl;
       //    prlong long(num);
            if(n<num)
            ans+=(i+1)%mod*(((n>>cnt)-1)%mod);
            else ans+=(i+1)%mod*((n>>cnt)%mod);
            ans=ans%mod;
        //    cout<<ans<<endl;
        }
        cout<<ans%mod<<endl;
    }
}