川大新生赛----羊工八刀----数学思维+前缀和

这题第一眼以为是差分,第二眼以为是dp,其实只是一个数学公式的化简

题目链接

对于每个含1的位置i,预处理出其前边所有位置的贡献就是
前i个数里,如果位置x是1,则加上(i-x)^2
公式化简就是 cnt1[i]*i*i+cnt3[i]-2*i*cnt2[i]

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
#define int long long
const int N = 1000010,mod = 1e9+7;
int f[N];
int cnt1[N],cnt2[N],cnt3[N];
signed main()
{
    int T,n;
    string s;
    cin>>T;
    while(T--)
    {
        memset(cnt1,0,sizeof cnt1);
        memset(cnt2,0,sizeof cnt2);
        memset(cnt3,0,sizeof cnt3);
        memset(f,0,sizeof f);
        cin>>n;
        cin>>s;
        s=" "+s;
        for(int i=1;i<s.size();i++)
        {
            cnt1[i]=cnt1[i-1]+(s[i]=='1');
            cnt2[i]=cnt2[i-1]+(s[i]=='1'?i:0);
            cnt3[i]=cnt3[i-1]+(s[i]=='1'?i*i%mod:0);
            if(s[i]=='1')
            f[i]=((cnt1[i]*i%mod*i%mod+cnt3[i]-2*i*cnt2[i]%mod)%mod+mod)%mod;
        }
        int res=0;
        for(int i=1;i<=n;i++)res=(res+f[i])%mod;
        cout<<res<<endl;
    }
}
上一篇:牛客寒假算法基础集训营


下一篇:P2078朋友