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;    }}