65922023-12-12 17:12:15Huni136Zenehallgatáscpp17Wrong answer 0/50254ms4688 KiB
#include <iostream>
#include <map>
#include <vector>
using namespace std;

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

// int binkeres(int x, int kezd, int veg)
// {
//     if(kezd<=veg)
//     {
//        int kozep=(kezd+veg)/2;
//        if(a[kozep]>=x && a[kozep-1]<x)
//        {
//             return kozep;
//        }
//        else if(a[kozep]<x && a[kozep+1]>=x)
//        {
//             return kozep+1;
//        }
//        else if(x>a[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[kozep]>=x && a[kozep-1]<x)
       {
            return kozep;
       }
       else if(a[kozep]<x && a[kozep+1]>=x)
       {
            return kozep+1;
       }
       else if(x>a[kozep])
       {
            //return binkeres(x,kozep+1,veg);
            kezd=kozep+1;
       }
       else
       {
            // return binkeres(x,kezd,kozep-1);
            veg=kozep-1;
       }
    }
    // int kozep;
    // while (kezd<=veg)
    // {
    //     if(veg-kezd==1)
    //     {
    //         if(x>a[kezd])
    //         {
    //             return veg;
    //         }
    //         else
    //         {
    //             return kezd;
    //         }
    //     }
    //     else
    //     {
    //         kozep=(kezd+veg)/2;
    //         if(x==a[kozep])
    //         {
    //             return kozep;
    //         }
    //         else if(x<a[kozep])
    //         {
    //             veg=kozep;
    //         }
    //         else
    //         {
    //             kezd=kozep;
    //         }
    //     }
    // }
    

}

// 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[kozep])
//         {
//             //cout<<"alma";
//             cout<<kozep<<" ";
//             return;
//         }
//         else if(x<a[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[1];
    for(int i=2;i<=n;i++)
    {
        cin>>x;
        a[i]=a[i-1]+x;
        //vlmi[prevveg]=i;
    }
    int k;
   // auto seg=vlmi.end();
    //prevveg++;
    int hossz=a[n];
    for(int i=1;i<=kdb;i++)
    {
        cin>>k;
        k=k%(hossz);
        if(k==0)
        {
            cout<<n<<" ";
            continue;
        }
        if(k<a[1])
        {
            cout<<1<<" ";
        }
        else if(k>a[n-1])
        {
            //cout<<11111;
            cout<<n<<" ";
        }
        else
        {
            int bal=0, jobb=n-1;

            bool talalt=false;

            while (bal<=jobb && !talalt)
            {
                //cout<<"bal "<<bal<<" jobb "<<jobb<<" *** ";


                int kozepe=(bal+jobb)/2;
                if(bal+1==jobb)
                if(x<=a[bal])
                {
                    cout<<bal+1<<" ";
                    talalt=true;
                }
                else
                {
                    cout<<jobb+1<<" ";
                    talalt=true;
                }
                else if(x<a[kozepe])
                    jobb=kozepe;
                else
                    bal=kozepe;

            }
        }
        
    }
}
SubtaskSumTestVerdictTimeMemory
base0/50
1Wrong answer0/03ms2392 KiB
2Time limit exceeded0/0217ms2552 KiB
3Wrong answer0/24ms2768 KiB
4Wrong answer0/26ms2980 KiB
5Wrong answer0/24ms3208 KiB
6Wrong answer0/34ms3416 KiB
7Wrong answer0/34ms3784 KiB
8Wrong answer0/34ms3856 KiB
9Wrong answer0/2195ms3912 KiB
10Time limit exceeded0/2202ms3792 KiB
11Wrong answer0/2193ms3792 KiB
12Time limit exceeded0/2217ms3928 KiB
13Time limit exceeded0/2226ms4088 KiB
14Time limit exceeded0/2201ms4220 KiB
15Time limit exceeded0/2203ms4164 KiB
16Wrong answer0/2187ms4300 KiB
17Wrong answer0/2184ms4228 KiB
18Wrong answer0/2185ms4324 KiB
19Time limit exceeded0/2243ms4304 KiB
20Wrong answer0/2180ms4308 KiB
21Time limit exceeded0/2254ms4424 KiB
22Wrong answer0/3180ms4472 KiB
23Time limit exceeded0/3244ms4560 KiB
24Wrong answer0/3180ms4688 KiB