2020牛客寒假算法基础集训营3 F 牛牛的Link Power I

题目描述

牛牛有一颗大小为n的神奇Link-Cut 数组，数组上的每一个节点都有两种状态，一种为link状态，另一种为cut状态。数组上任意一对处于link状态的无序点对（即(u,v)和(v,u)被认为是同一对）会产生dis(u,v)的link能量，dis(u,v)为数组上u到v的距离。 我们定义整个数组的Link能量为所有处于link状态的节点产生的link能量之和。 一开始数组上每个节点的状态将由一个长度大小为n的01串给出，'1’表示Link状态，'0’表示Cut状态。 牛牛想要知道整个数组的Link能量，为了避免这个数字过于庞大，你只用输出答案对10^9+7取余后的结果即可。

输入描述:

第一行输入一个正整数n(1≤n≤105)接下里一行输入一个长度大小为n的01串表示数组的初始状态，'1’表示Link状态，'0’表示Cut状态。

输出描述:

仅一行，表示整个数组的Link能量对10^9+7取余后的结果。

思维题，跟cf上之前做过的一道题很像，一开始没取模，过了百分之八十，以为自己写错了，好亏

后来取模就过了；
AC代码

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
string str;
int main()
{
    int n;
    cin>>n;
     cin>>str;
  
     int now = 0 , a = 0, y = 0;
   
     for(int  i = 0; i < str.size() ;i++ )
  
      {
       if ( str[i] == '1' )
   
      {
          
           a ++ ;
          
           now +=y;
           
           now%=mod;
           
           y+=a; 
           
           y%=mod;
       }
      if ( str[i] == '0' )
       {
           y += a;
           
            y%=mod;
       }
       }
       cout<< now%mod <<endl;
}