242612026-02-07 16:18:45ercseferencSzitakötő (50 pont)cpp17Runtime error 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;
}
SubtaskSumTestVerdictTimeMemory
base47/50
1Accepted0/01ms500 KiB
2Accepted0/052ms2868 KiB
3Runtime error0/11ms316 KiB
4Accepted1/11ms500 KiB
5Wrong answer0/11ms316 KiB
6Accepted1/11ms316 KiB
7Accepted1/11ms316 KiB
8Accepted1/11ms316 KiB
9Accepted1/11ms316 KiB
10Accepted2/21ms316 KiB
11Accepted2/21ms316 KiB
12Accepted2/21ms316 KiB
13Accepted2/21ms316 KiB
14Accepted2/21ms316 KiB
15Accepted2/21ms508 KiB
16Accepted2/22ms316 KiB
17Accepted2/22ms316 KiB
18Accepted2/21ms508 KiB
19Accepted2/21ms316 KiB
20Accepted2/21ms316 KiB
21Wrong answer0/11ms316 KiB
22Accepted2/239ms2328 KiB
23Accepted2/241ms2584 KiB
24Accepted2/252ms2844 KiB
25Accepted2/250ms2868 KiB
26Accepted2/248ms2716 KiB
27Accepted2/223ms2100 KiB
28Accepted2/241ms2612 KiB
29Accepted2/237ms2296 KiB
30Accepted2/252ms2860 KiB
31Accepted2/243ms2844 KiB