MT2018-找1

找1

#include<bits/stdc++.h> 
#include <string> 
using namespace std;
const int N=1e9+7;
//遇到1就计数，遇到0就计算ans累加，走到结尾也要进行ans计算
int main( )
{
    string s;
    getline(cin,s);
    int len=0;
    int ans=0;

    for(int i=0;i<s.length();i++)
    {
       /*if((str[i]=='1'&&len==0) || (len>0 && str[i]=='1'))
                 {
                     len++;
                 }
       else if(str[i]=='0'&&len>0)//说明到这里断开
       {
           ans+=(1+len)*len/2;
           len=0;
       }*/
       if(s[i]=='1') len++;
       if(s[i]=='0'||i==s.length()-1)
       {
           ans+=(1+len)*len/2;
           ans%=N;
           len=0;
       }
    }
    cout<<ans<<endl;
    return 0;
}