标签:pre cnt frac int Awesome Substrings CF1270F sq define
题目
给定一个01串,问有多少连续子串中1的个数大于0且可以整除其长度。
题解
问题等价于\(r-l=k(pre[r]-pre[l])\),即\(r-k\cdot pre[r]=l-k\cdot pre[l]\)。假设固定一个值\(T\),当\(k<T\)时,可以枚举\(k\)统计有多少相同数对,时间复杂度\(O(nT)\)。
由于\(pre[r]-pre[l] =\frac{r-l}{k}\le\frac{n}{T}\),也就是说当\(k\ge T\)时,合法子串内1的个数不会太多,最多只有\(\frac{n}{T}\)个。此时可以固定\(l\),枚举子串中1的个数\(m\),就能确定\(r\)的范围。又\(r-l \equiv 0 \mod m\),所以就能确定合法的\(r\)有多少(注意要取\(k\ge T\))的\(r\)。时间复杂度\(O(n\frac{n}{T})\)
总时间复杂度\(O(nT+n\frac{n}{T})\),取\(T=\sqrt{n}\),最低时间复杂度为\(O(n\sqrt{n})\)。
#include <bits/stdc++.h>
#define endl '\n'
#define IOS std::ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define mp make_pair
#define seteps(N) fixed << setprecision(N)
typedef long long ll;
using namespace std;
/*-----------------------------------------------------------------*/
ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;}
#define INF 0x3f3f3f3f
const int N = 3e5 + 10;
const double eps = 1e-5;
char s[N];
int pre[N];
int main() {
IOS;
cin >> s + 1;
int n = 0;
for(int i = 1; s[i]; i++) {
n++;
pre[i] = pre[i - 1] + (s[i] == '1');
}
ll ans = 0;
int sq = sqrt(n + 0.5) + 1;
vector<int> cnt(n * sq + n + 10);
for(int k = 1; k <= sq; k++) {
vector<int> id;
for(int i = 0; i <= n; i++) {
ll d = i - 1ll * k * pre[i];
if(d + n * sq >= cnt.size()) continue;
cnt[d + n * sq]++;
id.push_back(d + n * sq);
}
for(auto d : id) {
int num = cnt[d];
ans += 1ll * num * (num - 1) / 2;
cnt[d] = 0;
}
}
for(int k = 1; k <= sq; k++) {
int l = 1, r = 1;
for(int i = 0; i < n; i++) {
while(l <= n && pre[l] - pre[i] < k) l++;
while(r <= n && pre[r] - pre[i] <= k) r++;
ans += max(0, (r - i - 1) / k - max(l - i - 1, sq * k) / k);
}
}
cout << ans << endl;
}
标签:pre,cnt,frac,int,Awesome,Substrings,CF1270F,sq,define 来源: https://www.cnblogs.com/limil/p/15257351.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。