G Out of Sorts（二分， 优化）

G.G Out of Sorts

题意：

​ 给一组序列，问用二分查找每一个数，有多少数能查询成功 (n <= 1e6)

思路：

​ n 才1e6， 暴力枚举也才 nlogn

Code：

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

const int N=1000005;
int x[N], ans;

signed main()
{
    int n,m,a,c;
    cin>>n>>m>>a>>c>>x[0];
    for(int i=1;i<=n;i++)
    {
        x[i] = (a*x[i-1] + c) % m;
    }
    for(int i=1;i<=n;i++)
    {
        int l=1, r=n, k=x[i];
        while(l <= r)
        {
            int md = (l+r)>>1;
            if(x[md] == k) { ans++; break; }
            if(k > x[md]) l = md+1;
            if(k < x[md]) r = md-1;
        }
    }
    cout<<ans<<endl;

    return 0;
}

​ 赛后又想了想，发现可以用 logn 的算法实现：去二分查所有的数，对于查到的数，能满足条件的，ans++。

​ 那么怎样才算满足条件？

​ 假设现在已经查到了x[mid] , 接下来要去查找区间 [l, mid-1] 和 [mid+1, r], 对于区间 [l, mid-1] 的查询，<x[mid] 才算有效的，对于区间 [mid+1, r] 的查询，>x[mid] 才算有效的，所以要更新上下限，具体操作看代码。

Code：

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

const int N=1000005;
int x[N], ans;
void binary(int l,int r,int mn,int mx)
{
	if(l > r || mx < mn) return ;
	int md = (l+r) >> 1;
	if(x[md] > mn && x[md] < mx) ans++;
	binary(md+1, r, max(mn, x[md]), mx);
	binary(l, md-1, mn, min(mx, x[md]));
}

signed main()
{
	//freopen("1.out","w",stdout);
    int n,m,a,c;
    cin>>n>>m>>a>>c>>x[0];
    for(int i=1;i<=n;i++)
    {
        x[i] = (a*x[i-1] + c) % m;
    }
	int mx=1e12;
	binary(1,n,-1,mx);
    cout<<ans<<endl;

    return 0;
}