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≤10^5)接下里一行输入一个长度大小为n的01串表示数组的初始状态，'1’表示Link状态，'0’表示Cut状态。
输出描述:

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

3
101
输出

2
示例2
输入

5
00110
输出

1
示例3
输入

6
000010
输出

0

#include
#include 
using namespace std;
typedef long long ll;
const int mod=1e9+7;
char a[100100];
ll b[100100],c[100100];
int main()
{
 	ll n,k=0;
 	cin>>n;
 	cin>>a;
 	ll sum=0;
 	for(ll i=0;i<n;i++)
  		if(a[i]=='1')
   			b[k++]=i;
 	for(ll i=1;i<k;i++)
  		c[i]=b[i]-b[i-1];
 	for(ll i=1;i<k;i++)
  		sum=(sum+c[i]*(k-i)*i)%mod;
 	cout<<sum;
 	return 0;
}

H 牛牛的k合因子数

题目描述
合数是指自然数中除了能被1和本身整除外，还能被其他数（0除外）整除的数。 牛牛最近在研究“k合因子数”，所谓“k合数”是指一个数的所有因子中，是合数的因子共有k个。
例如20的因子有1,2,4,5,10,20，其中4,10,20为合数，它有3个合数因子，就称20是一个 “3合因子数” 牛牛想要知道1~n中给定k的情况下k合因子数的数目。
输入描述:

第一行输入两个数字n,m(1≤n,m≤10^5)表示范围以及查询“k”的数目接下来m行，每行一个正整数k(1≤k≤n)查询k合因子数的数目。
输出描述:

一行一个数字，表示k合因子数的数目。
示例1
输入

10 5
1
2
3
4
5
输出

4
1
0
0
0
说明
1~10的范围内1合因子数有：4,6,9,10，共4个2合因子数有：8，共1一个