【2016江苏省小学生夏令营选拔】小 X 分砖块(brick)

(Input: brick.in, Output: brick.out)

时间限制: 1 s 空间限制: 256 MB

这还是昨天的考试题目！

题目描述

小 X 喜欢跟着爸爸跑到建筑工地上去。
这天，小 X 看到一排砖，每块要么是白色的（0），要么是黑色的（1）。小 X 想把这排砖分成若干非空段，使得每段白砖和黑砖块数的比例相同。
当然，小 X 可以直接把整排砖作为一段，那就太简单了。为了增加难度，小 X 想知道最多能分成多少段，例如：
100011 = 10 + 0011（即样例 1，最多分成 2 段，比例为 1:1）；
0001110000000001 = 0001 + 11000000 + 0001（即样例 2，最多分成 3 段，比例为 3:1）。
小 X 百思不得其解，希望你帮帮他。

输入

从文件 brick.in 中读入数据。

第一行包含一个整数 N。我们将用 N 行来描述这排砖，初始时这排砖为空。
接下来 N 行，每行包含用一个空格隔开的两个整数 Ki,Ci（Ci 只可能是 0 或 1），表示在上一行描述完后尾部又有了 Ki 块颜色为 Ci 的砖。

输出

输出到文件 brick.out 中。

第一行包含一个整数，表示最多能分成的段数。

样例输入 复制

样例输入1
3
1 1
3 0
2 1
样例输入2
4
3 0
3 1
9 0
1 1

样例输出 复制

样例输出1

2
样例输出2

3

数据范围限制

对于 30%的数据，N=1。
对于 60%的数据，所有 Ki 均相等。
对于 100%的数据，1≤N≤100000，1≤Ki≤1000000000，砖的总块数不超过 1000000000。

经过了长时间的钻研，终于做出来了：

#include<bits/stdc++.h>
using namespace std;
long long n,k[1000001],c[1000001],x=0,y=0,ans=0;
int main()
{
	freopen("brick.in","r",stdin);
	freopen("brick.out","w",stdout);
	cin>>n;
	for(long long i=1;i<=n;++i)
	{
		cin>>k[i]>>c[i];
		if(c[i]==0) 
        x+=k[i];
		if(c[i]==1) 
        y+=k[i];
	}
	if(x==0||y==0)
	{
		cout<<x+y;
		return 0;
	} 
	int u=min(x,y);
	for(int i=u/2;i>=2;--i)
	{
		if(x%i==0&&y%i==0)
		{
			x/=i;
			y/=i;
		}
	}
	int a=0,b=0;
	for(long long i=1;i<=n;++i)
	{
		if(!c[i])
		{
			if(b*x%y==0)
			{
				int z=b*x/y-a;
				if(z>=1&&z<=k[i]) 
                ans++;
			 }
			 a+=k[i];
		}
		if(c[i])
		{
			if(a*y%x==0)
			{
				int z=a*y/x-b;
				if(z>=1&&z<=k[i]) 
                ans++;
			} 
			b+=k[i];
	 	}
	}
	cout<<ans;
}