bzoj3613: [Heoi2014]南園滿地堆輕絮(二分+貪心)

Hanks_o發表於2018-04-12

題目傳送門

解法:
二分答案。
因為後面的元素要大於等於前面的元素。
那麼我們使得前面的元素在滿足答案的情況下儘量的小就行。
這樣後面的元素選擇的餘地就更大。

貪心一下咯。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
ll a[5100000],A,B,C,D,mod;
int n;
ll pow_mod(ll a,int b) {
    ll ans=1ll;a%=mod;
    while(b!=0) {
        if(b%2==1)ans=(ans*a)%mod;
        a=(a*a)%mod;b/=2;
    }return ans;
}
ll F(ll x) {
    return ((A*pow_mod(x,3))%mod+(B*pow_mod(x,2))%mod+C*x%mod+D)%mod;
}
int main() {
    scanf("%d%lld%lld%lld%lld%lld%lldlld",&n,&A,&B,&C,&D,&a[1],&mod);
    a[0]=0ll;for(int i=2;i<=n;i++)a[i]=(F(a[i-1])+F(a[i-2]))%mod;
    ll l=0,r=mod-1,mid,ans;
    while(l<=r) {
        mid=(l+r)/2ll;
        ll sum=max(a[1]-mid,0ll);
        bool bk=true;
        for(int i=2;i<=n;i++) {
            if(a[i]>=sum)sum=max(sum,a[i]-mid);
            else {
                if(a[i]+mid<sum){bk=false;break;}
                sum=min(sum,a[i]+mid);
            }
        }if(bk==true) {r=mid-1;ans=mid;}
        else l=mid+1;
    }printf("%lld\n",ans);
    return 0;
}

相關文章