242612026-02-07 16:18:45ercseferencSzitakötő (50 pont)cpp17Futási hiba 47/5052ms2868 KiB
#include <bits/stdc++.h>
using namespace std;
using ll=long long;
int main()
{
    ll mod=1e9+7;
    int n,m; cin>>n>>m; m--;
    vector<ll>s(n);
    cin>>s[0];
    for(int i=1; i<n; i++){
        ll x; cin>>x;
        s[i]=s[i-1]+x;}
    ll t=s[m]-s[m-1];
    int poz=m-1;
    while(t<s[poz] && poz>=0){
        t+=s[poz]-s[poz-1];
        poz--;}
    ll ans=1;
    poz++;
    for(int i=0; i<poz; i++)
        ans=(ans*2)%mod;
    vector<ll>dp(n);
    dp[m-1]=0;
    dp[m]=ans;
    bool leh=1;
    for(int i=m+1; i<n; i++){
        int x=m+1,y=i;
        if(2*s[i-1]<s[i])leh=0;
        while(leh && x!=y){
            int k=(x+y)/2;
            if(2*s[k-1]>s[i])y=k;
            else x=k+1;}
        dp[i]=(2*dp[i-1]-dp[x-2])%mod;
        if(dp[i]<0)dp[i]+=mod;}
    if(leh){
        dp[n-1]-=dp[n-2];
        dp[n-1]=(dp[n-1]*2)%mod;
        if(dp[n-1]<0)dp[n-1]+=mod;
        cout<<dp[n-1];}
    else cout<<0;
    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base47/50
1Elfogadva0/01ms500 KiB
2Elfogadva0/052ms2868 KiB
3Futási hiba0/11ms316 KiB
4Elfogadva1/11ms500 KiB
5Hibás válasz0/11ms316 KiB
6Elfogadva1/11ms316 KiB
7Elfogadva1/11ms316 KiB
8Elfogadva1/11ms316 KiB
9Elfogadva1/11ms316 KiB
10Elfogadva2/21ms316 KiB
11Elfogadva2/21ms316 KiB
12Elfogadva2/21ms316 KiB
13Elfogadva2/21ms316 KiB
14Elfogadva2/21ms316 KiB
15Elfogadva2/21ms508 KiB
16Elfogadva2/22ms316 KiB
17Elfogadva2/22ms316 KiB
18Elfogadva2/21ms508 KiB
19Elfogadva2/21ms316 KiB
20Elfogadva2/21ms316 KiB
21Hibás válasz0/11ms316 KiB
22Elfogadva2/239ms2328 KiB
23Elfogadva2/241ms2584 KiB
24Elfogadva2/252ms2844 KiB
25Elfogadva2/250ms2868 KiB
26Elfogadva2/248ms2716 KiB
27Elfogadva2/223ms2100 KiB
28Elfogadva2/241ms2612 KiB
29Elfogadva2/237ms2296 KiB
30Elfogadva2/252ms2860 KiB
31Elfogadva2/243ms2844 KiB