CodeForces 628E Zbazi in Zeydabad

這個樹狀數組很巧妙。

將$O(n^3)$降到$O(n^2logn)$。

每個對角線的加和相等，建在一棵樹上。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3005;
int c[maxn << 1][maxn];
int l[maxn][maxn];
int dig[maxn][maxn];
int r[maxn][maxn];
char s[maxn][maxn];
int n, m;
typedef long long
 
 ll;
void add(int id, int x)
{
    while(x <= n)
    {
        c[id][x] += 1;
        x += (x & -x);
    }
}
int sum(int id, int x)
{
    int res = 0;
    while(x > 0)
    {
        res += c[id][x];
         x -= (x & -x);
    }
    return res;
}
int main()
{
    scanf("%d %d", &n, &m);
    
 
for(int i = 1; i <= n; i++)
    {
        scanf("%s", s[i] + 1);
    }
    /// O(nm)預處理出向左的最長，向左下的最長，還有是不是水平線段的最右點
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)
        {
            if(s[i][j] == ‘.‘)
            {
                l[i][j] = 0;
            }
            
 
else if(s[i][j] == ‘z‘)
            {
                l[i][j] = l[i][j - 1] + 1;
            }
            if(s[i][j + 1] != ‘z‘)
            {
                r[i][j] = 0;
            }
            else r[i][j] = 1;
        }
    }
    for(int i = n; i >= 1; i--)
    {
        for(int j = 1; j <= m; j++)
        {
            if(s[i][j] == ‘.‘)
            {
                dig[i][j] = 0;
            }
            else
            {
                dig[i][j] = dig[i + 1][j - 1] + 1;
            }
        }
    }
    ll ans = 0;
    for(int j = m; j >= 1; j--)
    {
        for(int i = 1; i <= n; i++)
        {
            if(s[i][j] == ‘z‘ && r[i][j] == 0)
            {
                int pos = j;
                while(s[i][pos] == ‘z‘)
                {
                    add(i + pos, i);
                    pos--;
                }
            }
        }
        for(int i = 1; i <= n; i++)
        {
            if(s[i][j] == ‘z‘) ///枚舉右上端點
            {
                ans += sum(i + j, i + min(l[i][j], dig[i][j]) - 1) - sum(i + j, i - 1);
            }
        }
    }
    printf("%lld\n", ans);
    return 0;
}

Grid Coloring

寫的傻逼樣，明天再改吧。

傻逼代碼：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int dp1[33][33][3];
int dp2[33][33][3];
char s[33][33];
int main()
{
    int n, m; scanf("%d %d", &n, &m);
    for(int i = 1; i <= n; i++)
    {
        scanf("%s", s[i] + 1);
    }
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)
        {
            if(s[i][j] == ‘B‘)
            {
                for(int p = 1; p <= i; p++)
                {
                    for(int q = 1; q <= j; q++)
                    {
                        if(s[p][q] == ‘.‘)
                        {
                            s[p][q] = ‘B‘;
                        }
                    }
                }
            }
        }
    }
    for(int i = n; i >= 1; i--)
    {
        for(int j = m; j >= 1; j--)
        {
            if(s[i][j] == ‘R‘)
            {
                for(int p = n; p >= i; p--)
                {
                    for(int q = m; q >= j; q--)
                    {
                        if(s[p][q] == ‘.‘)
                        {
                            s[p][q] = ‘R‘;
                        }
                    }
                }
            }
        }
    }
    for(int i = 1; i <= n; i++) printf("%s\n", s[i] + 1);
    if(s[1][1] == ‘B‘ || s[1][1] == ‘.‘) dp1[1][1][0] = 1;
    if(s[n][m] == ‘R‘ || s[n][m] == ‘.‘) dp2[n][m][1] = 1;
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)=
        {
            if(i == 1 && j == 1) continue;
            if(s[i][j] == ‘B‘)
            {
                ///那我就檢查是否合法
                int flag = 0;
                if(j > 1 && dp1[i][j - 1][0] == 0) flag = 1;
                if(i > 1 && dp1[i - 1][j][0] == 0) flag = 1;
                if(!flag) dp1[i][j][0] = 1;
                else dp1[i][j][0] = 0;
            }
        }
    }
    for(int i = n; i >= 1; i--)
    {
        for(int j = m; j >= 1; j--)
        {
            if(i == n && j == m) continue;
            if(s[i][j] == ‘R‘ || s[i][j] == ‘.‘)
            {
                int flag = 0;
                if(i < n && dp2[i + 1][j][1] == 0) flag = 1;
                if(j < m && dp2[i][j + 1][1] == 0) flag = 1;
                if(!flag) dp2[i][j][0] = 1;
                else dp2[i][j][0] = 0;
            }
        }
    }
    int ans = 0;
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)
        {

        }
    }
    printf("%d\n", ans);
    return 0;
}
/*
7 6
......
.....B
.B..R.
......
...B..
......
.R.R..
*/

打卡1