洛谷 P1795 无穷的序列_NOI导刊2010提高（05）

P1795 无穷的序列_NOI导刊2010提高（05）

题目描述

有一个无穷序列如下：

110100100010000100000…

请你找出这个无穷序列中指定位置上的数字

输入输出格式

输入格式：

第一行一个正整数N，表示询问次数；

接下来的N行每行一个正整数Ai，Ai表示在序列中的位置。

输出格式：

N行，每行为0或l，表示序列第Ai位上的数字。

输入输出样例

输入样例#1： 复制

4
3
14
7
6

输出样例#1： 复制

0
0
1
0

说明

对于100%的数据有N≤1500000，Ai≤10^9

思路：前缀和+二分。

#include<map>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
map<int,bool>ma;
int n,x,sum=;
int main(){
    scanf("%d",&n);
    for(int i=;i;i++){
        ma[sum]=,sum+=i;
        if(sum>)    break;
    }
    while(n--){
        scanf("%d",&x);
        if(ma[x])    printf("1\n");
        else printf("0\n");
    }
}

STL TLE 90分

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int ma[];
int l,r,mid;
int n,x,sum=;
int main(){
    scanf("%d",&n);
    ma[]=;
    for(int i=;i<=;i++)
        ma[i]=ma[i-]+i-;
    while(n--){
        scanf("%d",&x);
        l=;r=;int flag=;
        for(int i=;i<=;i++){
            mid=(l+r)/;
            if(ma[mid]<x)    l=mid+;
            else if(ma[mid]>x)    r=mid-;
            else if(ma[mid]==x){ flag=;break; }
        }
        if(flag)    printf("1\n");
        else printf("0\n");
    }
}