洛谷P4889 kls和flag（dp + 線性篩 + 數據結構）

2020.6.29
一上午腦子不在線，數據結構都寫不動，不知道爲什麼，是昨天睡得太死的問題麼？可能昨天週日去圖書館泡了一天，今天腦子就告訴我不要學習了吧？怎麼搞的，快正常起來把。我現在literally只想恰飯 + 躺。什麼也不想幹啊。

學校那邊也沒動靜了，我真的不抱什麼希望了，愛怎麼辦怎麼辦吧，別坑我錢就好。今天下午就不學了，錄一期vlog總結一下這魔幻的一年吧。

這道題我是搜概率統計搜出來的，就是問左倒和右倒的能覆蓋的端點，那自然很好想啊，是個錘子概率哦。暴力N方百分百過不去，30分，就不費力氣了。後來仔細看看，就是做一個線性篩dp那樣子。btw學了dp看啥都像dp。然後統計一下左和右的最大的距離，就是下標加減杆子長度上以前累加的數量加上去，然後再+1就行了。然後re，555555.

看了一下，很多人用map，但是我覺得這樣做是賭出題人不出數據卡吧，如果我是出題人，我肯定會把1e9拉滿，然後就re了。今天日記寫的有點敷衍，明天繼續去圖書館學習。

代碼：

#include <bits/stdc++.h>
using namespace std;
#define limit (1000000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步兩步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(ll i = a; i <= b ; ++i)
#define per(i, a, b) for(ll i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 988244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
#define FOUT freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\dabiao.txt", "wt", stdout)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快讀
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}
ll n,k;
struct node{
    ll val, x;
    bool operator<(const node &rhs)const{
        return val < rhs.val;
    }
}a[limit];
map<int, int> dp;
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    n = read(),k = read();
    rep(i ,1,n){
        a[i].val = read();
        a[i].x = i;
    }
    ll ans = 0;
    rep(i ,1, n){
        int minn = a[i].x - a[i].val, maxx = a[i].x + a[i].val;
        ans += dp[minn] + dp[maxx];
        ++dp[minn];
        ++dp[maxx];
    }
    write(ans);
    return 0;
}