JPEG 解码 CCF202212-3 模拟 题解

题目链接: oj

解题思路

就是简单的模拟，不要害怕无敌长的题干和数学公式。
要注意的是，再经过（7,0）点后，扫描的顺序发生了改变。

AC代码

#include <bits/stdc++.h>
using namespace std;
#define pi acos(-1)
int q[8][8], m[8][8] = {0};
int res[8][8];
double a(int u)
{
    if (u == 0)
    {
        return sqrt(0.5);
    }
    else
    {
        return 1;
    }
}
int main()
{
    for (int i = 0; i < 8; i++)
    {
        for (int j = 0; j < 8; j++)
        {
            scanf("%d", &q[i][j]);
        }
    }
    int n, t, i = 0, j = 0, k = 0;
    scanf("%d%d", &n, &t);
    while (n)
    {
        if (i == 7 && j == 0)
            break;
        if (k == 0)
        {

            scanf("%d", &m[i][j]);
            n--;
            if (!n)
                break;
            if (j != 7)
                j++;
            k++;
        }
        else if (k == 1)
        {
            while (i < 7 && j > 0)
            {
                scanf("%d", &m[i][j]);
                n--;
                if (!n)
                    break;
                i++;
                j--;
            }
            k++;
        }
        else if (k == 2)
        {

            scanf("%d", &m[i][j]);
            n--;
            if (!n)
                break;

            i++;
            k++;
        }
        else if (k == 3)
        {
            while (i > 0 && j < 7)
            {
                scanf("%d", &m[i][j]);
                n--;
                if (!n)
                    break;
                i--;
                j++;
            }
            k = 0;
        }
    }
    k = 0;
    while (n)
    {
        if (k == 0)
        {

            scanf("%d", &m[i][j]);
            n--;
            if (!n)
                break;
            if (j != 7)
                j++;
            k++;
        }
        else if (k == 3)
        {
            while (i < 7 && j > 0)
            {
                scanf("%d", &m[i][j]);
                n--;
                if (!n)
                    break;
                i++;
                j--;
            }
            k = 0;
        }
        else if (k == 2)
        {

            scanf("%d", &m[i][j]);
            n--;
            if (!n)
                break;

            i++;
            k++;
        }
        else if (k == 1)
        {
            while (i > 0 && j < 7)
            {
                scanf("%d", &m[i][j]);
                n--;
                if (!n)
                    break;
                i--;
                j++;
            }
            k++;
        }
    }
    if (t == 0)
    {
        for (int i = 0; i < 8; i++)
        {
            for (int j = 0; j < 8; j++)
            {
                printf("%d ", m[i][j]);
            }
            printf("\n");
        }
        return 0;
    }

    for (i = 0; i < 8; i++)
    {
        for (j = 0; j < 8; j++)
        {
            m[i][j] = m[i][j] * q[i][j];
        }
    }
    if (t == 1)
    {
        for (int i = 0; i < 8; i++)
        {
            for (int j = 0; j < 8; j++)
            {
                printf("%d ", m[i][j]);
            }
            printf("\n");
        }
        return 0;
    }
    int u, v;
    for (i = 0; i < 8; i++)
    {
        for (j = 0; j < 8; j++)
        {
            double sum = 0;
            for (u = 0; u < 8; u++)
            {
                for (v = 0; v < 8; v++)
                {
                    sum += a(u) * a(v) * m[u][v] * cos(pi / 8 * u * (i + 0.5)) * cos(pi / 8 * v * (j + 0.5));
                }
            }
            int ans = round(sum / 4 + 128);
            res[i][j] = max(min(ans, 255), 0);
        }
    }
    for (int i = 0; i < 8; i++)
    {
        for (int j = 0; j < 8; j++)
        {
            printf("%d ", res[i][j]);
        }
        printf("\n");
    }
    return 0;
}