AcWing 2872. 子串分值和

原题链接

考察:思维

思路:

        和CF817D一样的思路,考虑单点对答案的贡献.这题没必要统计左右边界.只需要记录上一个与当前字符相同的位置,右边界是字符串最右边.详情见代码

 1 #include <iostream> 
 2 #include <cstring>
 3 using namespace std;
 4 typedef long long LL;
 5 const int N = 100010,M = 30;
 6 char s[N];
 7 int len,L[M],R[M],vis[M];
 8 void init()
 9 {
10     for(int i=1;i<=len;i++)
11     {
12         int idx = s[i]-'a'+1;
13         L[i] = vis[idx];
14         vis[idx] = i;
15     }
16     for(int i=1;i<=28;i++) vis[i] = len+1;
17     for(int i=len;i>=1;i--)
18     {
19         int idx = s[i]-'a'+1;
20         R[i] = vis[idx];
21         vis[idx] = i;
22     }
23 }
24 void solve()
25 {
26     LL ans = 0;
27     memset(vis,0,sizeof vis);
28     for(int i=len;i>=1;i--)
29     {
30         int idx = s[i]-'a'+1;
31         if(!vis[idx]) vis[idx] = i;
32     }
33     for(int i=1;i<=len;i++)
34     {
35         int idx = s[i]-'a'+1;
36         if(vis[idx]!=i) ans+=(LL)(i-L[i])*(len+1-i);
37         else ans+=(LL)(i-L[i])*(R[i]-i);
38     }
39     printf("%lld\n",ans);
40 }
41 int main()
42 {
43     scanf("%s",s+1);
44     len = strlen(s+1);
45     init();
46     solve();
47     return 0;
48 }

 

posted @ 2021-04-13 22:59  acmloser  阅读(56)  评论(0编辑  收藏  举报