标签:位置 const int res sum CF1428F pos Fruit Sequences
题目链接https://codeforces.com/contest/1428/problem/F
参考题解https://www.cnblogs.com/zkyJuruo/p/13833960.htmlOI爷,永远滴神!
思路
对于01字符串从右向左遍历,如果当前\(s[i]=0\),那么对序列没有任何贡献,如果\(s[i]=1\),如果出现连续的1,设在第i位该值为\(h[i]\),那么从当前位到上一次出现\(h[i]\)之间的值,需要全部加上1 。例举一个样例就是110011,如果此时\(i=2\),那么从\([2,4]\)这一段范围,它的结果就会+3,而\([5,6]\)的代价还是不变的。如果\(i=1\),那么从\([1,5]\)的范围之内代价都会+1,结果就是+5。由此看来就只要看当前出现的\(h[i]\)的这个值,在上一次所出现的位置在哪里,那么代价就会增加\(h[i]-i\),如果不存在这样的位置,那么代价就相当于直接加上\(n-i+1\)。
开一个pos数组记录每一个h[i]最后出现的位置。因为在连续求1的过程中,其最后的位置恰好是反一反的,例如110011,h[5]=2,h[6]=1,但是在遍历到位置2的时候,h[2]=1所对应1应该是第5位,h[1]=2所对应的位置应该是第6位,所以在发现s[i]=0 && s[i+1]=1的时候,要把pos数组的一部分reverse一下。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define int LL
typedef pair<int, int> PII;
const int inf = 0x3f3f3f3f;
const int N = 5e5 + 10;
const int mod = 1e9 + 7;
char s[N];
int pos[N], h[N];
void solve() {
int n;
scanf("%lld%s", &n, s + 1);
int res = 0, sum = 0;
for(int i = n; i; i--) {
h[i] = s[i] == '0' ? 0 : h[i + 1] + 1;
if(s[i] == '1') {
if(!pos[h[i]]) sum += n - i + 1;
else sum += pos[h[i]] - i;
pos[h[i]] = i;
}
else if(s[i] == '0' && s[i + 1] == '1') {
int x = h[i + 1];
reverse(pos + 1, pos + 1 + x);
}
res += sum;
// printf("i = %d sum = %d\n", i, sum);
}
printf("%lld\n", res);
}
signed main() {
// freopen("in.txt", "r", stdin);
// int t; cin >> t; while(t--)
solve();
return 0;
}
标签:位置,const,int,res,sum,CF1428F,pos,Fruit,Sequences 来源: https://www.cnblogs.com/ZX-GO/p/13904077.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。