242662026-02-07 16:33:37ercseferencSzitakötő (50 pont)cpp17Wrong answer 49/5052ms1968 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--;
    if(n==3){
        ll x,y,z; cin>>x>>y>>z;
        if(m==0)cout<<0;
        if(m==1)if(x+y>z)cout<<4;else cout<<0;
        if(m==2)if(x+y<=z)cout<<8; else cout<<4;
        return 0;}
    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
base49/50
1Accepted0/01ms316 KiB
2Accepted0/052ms1844 KiB
3Accepted1/11ms316 KiB
4Accepted1/11ms316 KiB
5Accepted1/11ms316 KiB
6Accepted1/11ms316 KiB
7Accepted1/11ms316 KiB
8Accepted1/11ms316 KiB
9Accepted1/11ms316 KiB
10Accepted2/21ms316 KiB
11Accepted2/21ms500 KiB
12Accepted2/21ms316 KiB
13Accepted2/21ms508 KiB
14Accepted2/21ms508 KiB
15Accepted2/21ms316 KiB
16Accepted2/21ms332 KiB
17Accepted2/22ms316 KiB
18Accepted2/22ms508 KiB
19Accepted2/22ms508 KiB
20Accepted2/22ms316 KiB
21Wrong answer0/11ms316 KiB
22Accepted2/239ms1844 KiB
23Accepted2/241ms1844 KiB
24Accepted2/252ms1964 KiB
25Accepted2/250ms1844 KiB
26Accepted2/246ms1968 KiB
27Accepted2/223ms1964 KiB
28Accepted2/241ms1848 KiB
29Accepted2/235ms1964 KiB
30Accepted2/250ms1844 KiB
31Accepted2/243ms1964 KiB