【贪心·思维·细节】jzoj4879 少女觉 纪中集训提高B组

4879 少女觉
(File IO): input:silly.in output:silly.out
Time Limits: 1000 ms Memory Limits: 262144 KB Detailed Limits

Description

在幽暗的地灵殿中，居住着一位少女，名为古明地觉。
据说，从来没有人敢踏入过那座地灵殿，因为人们恐惧于觉一族拥有的能力——读心。
掌控人心者，可控天下。

咳咳。
人的记忆可以被描述为一个黑块(B)与白块(W)的序列，其中情感值被定义为序列中黑块数量与白块数量之比。
小五口在发动读心术时，首先要解析人的记忆序列，因此，需要将序列分割为一些段，并且要求每一段记忆序列的情感值都相等。
下面给出两个例子：
BWWWBB -> BW + WWBB (Ratio=1:1)
WWWBBBWWWWWWWWWB -> WWWB + BBWWWWWW + WWWB (Ratio=3:1)
现在小五手上有一个人的记忆序列，她想要知道，如何将手中的记忆序列分成尽可能多的段呢？

Input

第一行包含一个正整数T，代表数据组数。
对于每一组测试数据，第一行包含一个正整数N。
接下来N行描述一个序列，每行包含一个正整数K和一个大写字母C，表示序列接下来有连续K个颜色为C的方块。

Output

对于每组测试数据输出一行一个正整数，表示最多分成的段数。

Sample Input

3
3
1 B
3 W
2 B
4
3 W
3 B
9 W
1 B
2
2 W
3 W

Sample Output

2
3
5

Data Constraint

对于10%的数据，n<=15
对于20%的数据，n<=500
另有30%的数据，K=1
另有30%的数据，K<=50
对于100%的数据，$N<=10^5，序列长度不超过10^9$
保证对于全部测试点，输入文件行数不超过$2.5\ast 10^6$

---

考试的时候想到了正解，但是由于各种奇葩的细节错误（而且这道题还是多组数据）导致爆零。
首先可以知道，它的比例是一定的，等于所有黑色数量和所有白色数量的比例，所以我们把这个比例算出来，然后如果碰到一组满足比例就ans++

/*
思路就是正解的思路 其实这道题超简单哇
至于为什么会写挂呢。。。
continue写成了return 0;
还有就是我打代码的时候脑子里一直都是想的理想化的状态
就是黑白是交替进行的 然而并不是这样
就没有判断nd==0(另外那个颜色还没有 就只能把当前颜色选完)和nd为负数的情况(这个时候也只能把当前颜色选完)
只有在选部分当前颜色 可以达到那个比例的时候才选一部分 
*/
#include<queue>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 100005
#define INF 0x3f3f3f3f
#define ll long long
struct node{
	int k;char c;
}a[N];
int n,t1,t2;
inline int rd()
{
	int f=1,x=0;char c=getchar();
	while(c<'0'||'9'<c){if(c=='-')f=-1;c=getchar();}
	while('0'<=c&&c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return f*x;
}
int gcd(int a,int b)
{
	if(b==0) return a;
	return gcd(b,a%b);
}
int main()
{
	freopen("silly.in","r",stdin);
	freopen("silly.out","w",stdout);
	int T;
    T=rd();
    while(T--)
    {
    	n=rd();
    	t1=0,t2=0;
    	for(int i=1;i<=n;i++)
    	{
    		a[i].k=rd();a[i].c=getchar();
    		if(a[i].c=='W') t1+=a[i].k;
    		if(a[i].c=='B') t2+=a[i].k;
		}
		if(t1==0){printf("%d\n",t2); /*return 0;*/continue;}//这里是在干嘛啊喂！ 
		if(t2==0){printf("%d\n",t1); /*return 0;*/continue;}
		int d=gcd(t1,t2);
		t1/=d,t2/=d;
		int i=1,ans=0,num=0/*第i段用了的块数*/,n1=0/*已有的块数*/,n2=0;
		while(i<=n)
		{
			if(n1==0&&n2==0)
			{
				if(a[i].c=='W') n1+=a[i].k-num;
				if(a[i].c=='B') n2+=a[i].k-num;
				num=0;
				i++;
			}
			else
			{
				if(a[i].c=='W')
				{
					if(n2%t2||!n2)
					{
						n1+=a[i].k-num;
						num=0;
						i++;
						continue;
					}
					else
					{
						int nd=n2/t2*t1;
						nd=nd-n1;//要减去已经有了的才是真正需要的 
						if(a[i].k-num>=nd&&nd>=0/*有可能对于另一种颜色来说 这个颜色已经太多了超比例*/)
							ans++,n1=0,n2=0,num=nd;
						else n1+=a[i].k-num,i++,num=0;
					}
				}
				if(a[i].c=='B')
				{
					if(n1%t1||!n1)
					{
						n2+=a[i].k-num;
						num=0;
						i++;
						continue;
					}
					else
					{
						int nd=n1/t1*t2;
						nd=nd-n2;
						if(a[i].k-num>=nd&&nd>=0)
							ans++,n1=0,n2=0,num=nd;
						else n2+=a[i].k-num,i++,num=0;
					}
				}
			}
			if(num==a[i].k)
				i++,num=0;
		}
		printf("%d\n",ans);
	}
    return 0;
}

看到题解有一份代码，写得更简洁，学习一下：

/*
觉得这种写法还要优秀一些
他不是像我那样分段了重新计数 而是整体来算
因为反正整体的比例也还是那个比例
而且余下的那些也会再加入 
只是碰到了那种可以断开的地方就ans++
而我是硬生生地把他断开了 就要多处理很多细节 
*/
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<iostream>
using namespace std;
const int N=1e5+15;
int T; 
int k[N];
char ch[N][5];
int calc(int x,int a,int b)
{
    if (1ll*x*b%a!=0) return -1;else return 1ll*x*b/a;
}
int main()
{
    scanf("%d",&T);
    while (T--)
    {
        int n,sb=0,sw=0;
        scanf("%d",&n);
        for (int i=1;i<=n;i++)
        {
            scanf("%d",&k[i]);
            scanf("%s",ch[i]);
            if (ch[i][0]=='B') sb+=k[i];
            if (ch[i][0]=='W') sw+=k[i];    
        }
        //printf("%d %d\n",sb,sw);
        if (!sb||!sw) {printf("%d\n",sb+sw);continue;}
        int s1=0,s2=0,ans=0;
        for (int i=1;i<=n;i++)
        {
            if (ch[i][0]=='B') 
            {
                int p=calc(s2,sw,sb);
                if (p>s1&&p<=s1+k[i]) ans++;
                s1+=k[i];
            }
            if (ch[i][0]=='W') 
            {
                int p=calc(s1,sb,sw);
                if (p>s2&&p<=s2+k[i]) ans++;
                s2+=k[i];
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}