65952023-12-12 17:21:09Huni136Zenehallgatáscpp17Time limit exceeded 38/50264ms5008 KiB
#include <iostream>
//#include <map>
#include <vector>
using namespace std;

//map<int,int> vlmi;
vector<int> a(100002);

// int binkeres(int x, int kezd, int veg)
// {
//     if(kezd<=veg)
//     {
//        int kozep=(kezd+veg)/2;
//        if(a.at(kozep)>=x && a[kozep-1]<x)
//        {
//             return kozep;
//        }
//        else if(a.at(kozep)<x && a[kozep+1]>=x)
//        {
//             return kozep+1;
//        }
//        else if(x>a.at(kozep))
//        {
//             return binkeres(x,kozep+1,veg);
//        }
//        else
//        {
//             return binkeres(x,kezd,kozep-1);
//        }
//     }
// }

int binkeres(int x, int kezd, int veg)
{
     int kozep;
    while(kezd<=veg)
    {
       kozep=(kezd+veg)/2;
       if(a.at(kozep)>=x && a.at(kozep-1)<x)
       {
            return kozep;
       }
       else if(a.at(kozep)<x && a.at(kozep+1)>=x)
       {
            return kozep+1;
       }
       else if(x>a.at(kozep))
       {
            //return binkeres(x,kozep+1,veg);
            kezd=kozep+1;
       }
       else
       {
            // return binkeres(x,kezd,kozep-1);
            veg=kozep-1;
       }
    }
}

// void binkeres(int x, int kezd, int veg)
// {
//     //if(x==1)
//     //{
//         //cout<<kezd<<" "<<veg<<endl;
//     //}
//     if(veg-kezd==1)
//     {
//         if(x>a[kezd])
//         {
//             cout<<veg<<" ";
//             return;
//         }
//         else
//         {
//             cout<<kezd<<" ";
//             return;
//         }
//     }
//     else
//     {
//         int kozep=(kezd+veg)/2;
//         if(x==a.at(kozep))
//         {
//             //cout<<"alma";
//             cout<<kozep<<" ";
//             return;
//         }
//         else if(x<a.at(kozep))
//         {
//             binkeres(x,kezd,kozep-1);
//         }
//         else
//         {
//             binkeres(x,kozep+1,veg);
//         }
//     }
// }

int main()
{
    int n,x,kdb;
    cin>>n>>kdb;
    //vector<int> a(n+1);
    //a[0]=0;
    cin>>a.at(1);
    for(int i=2;i<=n;i++)
    {
        cin>>x;
        a.at(i)=a.at(i-1)+x;
        //vlmi[prevveg]=i;
    }
    int k;
   // auto seg=vlmi.end();
    //prevveg++;
    int hossz=a.at(n);
    for(int i=1;i<=kdb;i++)
    {
        cin>>k;
        k=k%(hossz);
        if(k==0)
        {
            cout<<n<<" ";
            continue;
        }
        if(k<a.at(1))
        {
            cout<<1<<" ";
        }
        else if(k>a.at(n-1))
        {
            //cout<<11111;
            cout<<n<<" ";
        }
        else
        {

            cout<<binkeres(k,1,n)<<" ";
        }
        // //auto it=vlmi.begin();
        // while (it!=seg && it->first<k)
        // {
        //     it++;
        // }
        // //cout<<it->second<<" ";
    }
}
SubtaskSumTestVerdictTimeMemory
base38/50
1Accepted0/03ms2664 KiB
2Time limit exceeded0/0252ms2940 KiB
3Accepted2/24ms3056 KiB
4Accepted2/24ms3228 KiB
5Accepted2/24ms3360 KiB
6Accepted3/34ms3448 KiB
7Accepted3/34ms3540 KiB
8Accepted3/34ms3736 KiB
9Accepted2/2200ms4024 KiB
10Time limit exceeded0/2224ms3800 KiB
11Time limit exceeded0/2257ms2900 KiB
12Time limit exceeded0/2261ms2968 KiB
13Accepted2/2194ms4116 KiB
14Accepted2/2190ms4128 KiB
15Time limit exceeded0/2239ms4344 KiB
16Accepted2/2190ms4460 KiB
17Accepted2/2194ms4444 KiB
18Accepted2/2187ms4444 KiB
19Time limit exceeded0/2264ms3540 KiB
20Time limit exceeded0/2250ms4656 KiB
21Accepted2/2187ms4796 KiB
22Accepted3/3189ms4836 KiB
23Accepted3/3185ms4880 KiB
24Accepted3/3186ms5008 KiB