C. Diluc and Kaeya
输入一行字符串包含D或者K,可以将其分成几块,也可以不分,求前缀D和K比例相同,做多可以分成几块。
input
5
3
DDK
6
DDDDDD
4
DKDK
1
D
9
DKDKDDDDK
output
1 2 1
1 2 3 4 5 6
1 1 1 2
1
1 1 1 2 1 2 1 1 3
求出D和K的比例,将其化简,并记录其出现的次数。
#include<bits/stdc++.h>
using namespace std;
pair<int,int> tmp;
map<pair<int,int>,int> mp;
string s;
int main()
{
int p;
scanf("%d",&p);
while(p--)
{
mp.clear();
int t,i,j,d=0,k=0,x,y;
scanf("%d",&t);
cin>>s;
for(i=0;i<t;i++)
{
if(s[i]=='D')
d++;
else
k++;
int g=__gcd(k,d);
x=k/g,y=d/g;
tmp={x,y};
mp[tmp]++;
printf("%d ",mp[tmp]);
}
printf("\n");
}
}