题目
Theofanis 有一个字符串 s1s2…sn 和一个字符 c。他希望使用最少的操作次数使字符串的所有字符都等于 c。
在一次操作中,他可以选择一个数 x (1≤x≤n) 并且对于每个位置 i,其中 i 不能被 x 整除,用 c 替换 si。
求使所有字符等于 c 和他应该在操作中使用的 x-s 所需的最少操作次数。
输入
第一行包含一个整数 t (1≤t≤104)——测试用例的数量。
每个测试用例的第一行包含整数 n (3≤n≤3⋅105) 和一个小写拉丁字母 c——字符串 s 的长度和结果字符串应包含的字符。
每个测试用例的第二行包含一个由小写拉丁字母组成的字符串 s — 初始字符串。
保证所有测试用例的 n 总和不超过 3⋅105。
输出
对于每个测试用例,首先打印一个整数 m——使所有字符等于 c 所需的最少操作次数。
接下来,打印 m 个整数 x1,x2,…,xm (1≤xj≤n)——应该按照给定顺序使用的 x-s。
可以证明,在给定的约束下,答案总是存在的。如果有多个答案,打印任何一个。
题解思路
一开始想用最大的素数来更新答案 ,即再判断 这个素数的 倍数 有没有需要修改的,需要的话就 让第二大的素数来重复这个操作,知道里面没有需要修改的。
实际上只需要两个素数就能解决这些了,另一个素数一定不被第一个整除 ,
而第一个已经把第二个修改为正确答案了。
当然还有两种情况需要特判,无需修改和只需要修改一个(即这个数的倍数全为正确的)。
暴力枚举只需修改一个的情况。
这里时间复杂度只有nlog级别的,有点类似埃筛(nloglog)但是比埃筛慢
for (int i = 1 ; i <= n ; i++ )
{
if ( s[i] == ch[0] )
{
int ok = 1 ;
for (int j = i*2 ; j <= n ; j+= i )
{
if ( s[j] != ch[0] )
{
ok = 0 ;
break ;
}
}
if ( ok )
{
cout << 1 << "\n" ;
cout << i << "\n" ;
book = 0 ;
break ;
}
}
}
两种的情况之间找n和n-1即可。
AC代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 300100 ;
char s[N] ;
int main ()
{
ios::sync_with_stdio(false);
int T ;
cin >> T;
while ( T -- )
{
int n ,cnt = 0 ,t1 = 0 ;
char ch[2] ;
cin >> n >> ch ;
cin >> s + 1 ;
for (int i = 1 ; i <= n; i++ )
{
if ( s[i] != ch[0] )
cnt++ ;
}
if ( cnt == 0 )
{
cout << "0\n" ;
continue;
}
int book = 1 ;
for (int i = 1 ; i <= n ; i++ )
{
if ( s[i] == ch[0] )
{
int ok = 1 ;
for (int j = i*2 ; j <= n ; j+= i )
{
if ( s[j] != ch[0] )
{
ok = 0 ;
break ;
}
}
if ( ok )
{
cout << 1 << "\n" ;
cout << i << "\n" ;
book = 0 ;
break ;
}
}
}
if ( book )
{
cout << 2 << "\n" ;
cout << n-1 << " " << n <<"\n";
}
}
return 0 ;
}