1. 程式人生 > 其它 >codeforces 1168 B. Good Triple(思維+暴力)

codeforces 1168 B. Good Triple(思維+暴力)

傳送門

題意:

給出一個長度為\(n\)\(01\)串,若一個區間滿足以下條件,則是合法的,

存在整數\(x,k\) 滿足 \(l\leq x <x+2k \leq r\) \(s_x=s_{x+k}=s_{x+2k}\)

求有多少區間是合法的

題解:

看完這道題感覺一點思路都沒有,於是去看了題解,大受震撼

事實上,一個長度為\(9\)的區間內必定存在一個滿足條件的位置,因此直接暴力找即可。

至於為什麼是長度為\(9\)之內必定有,可能是因為\(01\)串的關係,所有使得上述情況很容易出現。

由這題可以告訴我們,有些沒有思路的題,可以嘗試先暴力試下,說不定就是正解。

程式碼:

#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int MAXN=5e5+5;
const int inf=0x3f3f3f3f;
char s[MAXN];
int main()
{
    cin >> s + 1;
    int n = strlen(s + 1);
    int r = n + 1;
    ll ans = 0;
    for (int i = n - 2; i >= 1;i--)
    {
        for (int j = 1; i + 2 * j <= n;j++)
        {
            if(s[i]==s[i+j]&&s[i]==s[i+2*j]){
                r = min(r, i+2*j);
                break;
            }
        }
        
        if(r<=n){
            ans = ans + (n - r + 1);
        }
    }
    cout << ans << endl;
}