Dividing a String
Description
给出一个字符串S,将其分成最多的字符串S1…Sk,使得Si!=Si+1
|S|<=2e5
Solution
显然分成的字符串的长度为1或2
设Fn表示前n个的答案,那么如果Sn!=Sn-1那么Fn=Fn-1+1否则Fn=Fn-3+2
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e5+5;
char s[N];
int n,f[N];
int main() {
scanf("%s",s+1);n=strlen(s+1);
fo(i,1,n)
if (s[i]==s[i-1]) f[i]=(i>=3)?f[i-3]+2:1;
else f[i]=f[i-1]+1;
printf("%d\n",f[n]);
return 0;
}
RGB Balls
Description
有3N个球,颜色分别为’R’,‘G’,‘B’,每种颜色的球都有N个
有N个人,你需要给每个人分配每种颜色的球各一个
设第i个人被分到的是第ai,bi,ci个球,满足ai<bi<ci
你需要最小化∑ci-ai,问方案数
Solution
最小化的话直接贪心
维护每种组合RG,GB,RB,R,G,B的个数直接算
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=3e5+5,Mo=998244353;
int n,r,g,b,rg,rb,gb;
char s[N];
int main() {
scanf("%d",&n);
scanf("%s",s+1);
int ans=1;
fo(i,1,3*n) {
if (s[i]=='R') {
if (gb) {ans=(ll)ans*gb%Mo;gb--;continue;}
if (g) {ans=(ll)ans*g%Mo;rg++;g--;continue;}
if (b) {ans=(ll)ans*b%Mo;rb++;b--;continue;}
r++;
}
if (s[i]=='G') {
if (rb) {ans=(ll)ans*rb%Mo;rb--;continue;}
if (r) {ans=(ll)ans*r%Mo;rg++;r--;continue;}
if (b) {ans=(ll)ans*b%Mo;gb++;b--;continue;}
g++;
}
if (s[i]=='B') {
if (rg) {ans=(ll)ans*rg%Mo;rg--;continue;}
if (r) {ans=(ll)ans*r%Mo;rb++;r--;continue;}
if (g) {ans=(ll)ans*g%Mo;gb++;g--;continue;}
b++;
}
}
fo(i,1,n) ans=(ll)ans*i%Mo;
printf("%d\n",ans);
return 0;
}
Numbers on a Circle
Description
有一个大小为n的环,每次操作你可以选择相邻3个数a,b,c,把b变为a+b+c
给出两个环A和B,问将A变为B的最小操作次数
n<=2e5,A<=1e9
Solution
倒过来,变成选择a,b,c,将b减去a+c
那么显然b>a,b>c,并且直到b=b%(a+c)为止都不能操作a和c
直接用一个堆存所有元素,每次选择最大的操作
Code
#include <set>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
int read() {
char ch;
for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
int x=ch-'0';
for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x;
}
const int N=2e5+5;
#define L(x) (x==1?n:x-1)
#define R(x) (x==n?1:x+1)
set< pair<int,int> > s;
int n,a[N],b[N];
int main() {
n=read();
fo(i,1,n) a[i]=read();
fo(i,1,n) b[i]=read();
fo(i,1,n) s.insert(make_pair(-b[i],i));
ll ans=0;
while (!s.empty()) {
pair<int,int> tmp=*s.begin();s.erase(s.begin());
int x=tmp.second,cnt=b[L(x)]+b[R(x)];
if (b[x]==a[x]) continue;
if (b[x]<a[x]) {ans=-1;break;}
int d=b[x]-a[x];
if (d<cnt) {ans=-1;break;}
ans+=d/cnt;b[x]-=d/cnt*cnt;
s.insert(make_pair(-b[x],x));
}
printf("%lld\n",ans);
return 0;
}
Sorting a Grid
Description
给出一个n*m的矩阵A,每个位置的数是1~nm的排列
将A的每一行重排得到B
将B的每一列重排得到C
将C的每一行重排得到D
需要D[i][j]=(i-1)*m+1
给出一种方案
n,m<=100
Solution
C的每一行必须是[(i-1)m+1,im]
那么B的每一列必须要包含C的每一行的数各一个
构造二分图直接匹配,根据HALL定理显然有最大匹配
方案随便造
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define rep(i,a) for(int i=lst[a];i;i=nxt[i])
using namespace std;
int read() {
char ch;
for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
int x=ch-'0';
for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x;
}
const int N=205,M=N*N<<1,inf=0x7fffffff;
int t[M],f[M],nxt[M],lst[N],l;
void add(int x,int y,int z) {
t[++l]=y;f[l]=z;nxt[l]=lst[x];lst[x]=l;
t[++l]=x;f[l]=0;nxt[l]=lst[y];lst[y]=l;
}
int n,m,a[N][N],b[N][N],c[N][N];
bool used[N][N];
int P(int x) {return (x-1)/m+1;}
int S,T,dis[N],q[N];
bool bfs() {
fo(i,S,T) dis[i]=-1;dis[S]=0;
int i=0,j=1;q[1]=S;
while (i<j)
rep(k,q[++i])
if (f[k]&&dis[t[k]]==-1) {
dis[t[k]]=dis[q[i]]+1;
q[++j]=t[k];
}
return dis[T]!=-1;
}
int dinic(int x,int y) {
if (x==T) return y;
int now=0;
rep(i,x)
if (f[i]&&dis[t[i]]==dis[x]+1) {
int k=dinic(t[i],min(y,f[i]));
y-=k;now+=k;f[i]-=k;f[i^1]+=k;
if (!y) break;
}
if (!now) dis[x]=-1;
return now;
}
int main() {
n=read();m=read();
fo(i,1,n) fo(j,1,m) a[i][j]=read();
S=0;T=n<<1|1;
fo(w,1,m) {
l=1;fo(i,S,T) lst[i]=0;
fo(i,1,n)
fo(j,1,m)
if (!used[i][j])
add(i,n+P(a[i][j]),1);
fo(i,1,n) add(S,i,1);
fo(i,1,n) add(i+n,T,1);
while (bfs()) dinic(S,inf);
int p=2;
fo(i,1,n)
fo(j,1,m)
if (!used[i][j]) {
if (!f[p]) {
used[i][j]=1;
b[i][w]=a[i][j];
}
p+=2;
}
}
fo(i,1,n) {
fo(j,1,m) printf("%d ",b[i][j]);
puts("");
}
fo(j,1,m) fo(i,1,n) c[P(b[i][j])][j]=b[i][j];
fo(i,1,n) {
fo(j,1,m) printf("%d ",c[i][j]);
puts("");
}
return 0;
}
Reversing and Concatenating
Description
给出一个长度为N的字符串S,进行k次如下操作:
令T为S的反串,ST为S和T的拼接
从ST中选择一个长度为N的字符串作为新的S
问进行k次操作后得到的字典序最小字符串
N<=5000,k<=1e9
Solution
令c为S中最小的字符
考虑S中末尾连续的c,若长度为l,那么一次操作可以让l*2
若这一段c不在末尾,则可以用一次操作将这一段c变到末尾
那么枚举每一段连续的c,答案前缀c的长度很好计算
剩下的求一个字典序最小即可
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e4+5;
int n,k;
char st[N],an[N],tmp[N];
void chk() {
fo(i,1,n) {
if (an[i]<tmp[i]) return;
if (tmp[i]<an[i]) {
fo(j,1,n) an[j]=tmp[j];
return;
}
}
}
int main() {
scanf("%d%d",&n,&k);
scanf("%s",st+1);
char c='z';
fo(i,1,n) c=min(c,st[i]);
fo(i,1,n) st[2*n-i+1]=an[i]=st[i];
if (st[n]==c) {
int pos=n,len=0,y=k;
for(;pos>=1&&st[pos]==c;pos--) len++;
for(;len<n&&y;y--) len<<=1;
if (len<n) {
fo(i,1,len) tmp[i]=c;
fo(i,len+1,n) tmp[i]=st[pos--];
} else fo(i,1,n) tmp[i]=c;
chk();
}
for(int l=n+1,r=n;l<=n*2;l=r+1) {
while (r<n*2&&st[r+1]==st[l]) r++;
if (st[l]==c) {
int len=0,pos=r,y=k-1;
for(;pos>=r-n+1&&st[pos]==c;pos--) len++;
for(;len<n&&y;y--) len<<=1;
if (len<n) {
fo(i,1,len) tmp[i]=c;
fo(i,len+1,n) tmp[i]=st[pos--];
} else fo(i,1,n) tmp[i]=c;
chk();
}
}
printf("%s\n",an+1);
return 0;
}
Counting of Subarrays
咕咕咕