小粉兔做麻辣兔头

[语言月赛 202308] 小粉兔做麻辣兔头

题目描述

粉兔喜欢吃麻辣兔头，麻辣兔头的辣度分为若干级，用数字表示，数字越大，兔头越辣。为了庆祝粉兔专题赛 #1 的顺利举行，粉兔要做一些麻辣兔头送给扶苏、Max、小 F 和 Z，但是，粉兔不知道这些麻辣兔头应该做成什么辣度，TA 决定使用投骰子的方法来决定。

粉兔使用的骰子不是标准的 $6$ 面骰子。粉兔的骰子，每一面上写有 $1\sim 6$ 中的一个数字，$1$ 的对面为 $2$，$3$ 的对面为 $4$，$5$ 的对面为 $6$。粉兔将 $N$ 个骰子摞成一个骰子塔，从下向上数第 $i$ 个骰子，向上的一面写的数字为 $P_i$。

粉兔决定将麻辣兔头的总辣度做成这个骰子塔暴露在空气中所有面上数字之和，请问，总辣度为多少？

输入格式

输入共两行。

输入的第一行为一个整数 $N$。

输入的第二行为 $N$ 个整数，第 $i$ 个为 $P_i$。

输出格式

输出一行一个整数，代表兔头总辣度。

样例 #1

样例输入 #1

6
1 2 3 4 5 6

样例输出 #1

90

提示说明

样例 1 解释

侧面上数字总和为 $84$，顶上数字为 $6$，加和得答案为 $90$。

数据规模与约定

• 对于 $20\%$ 的数据，$N=1$；
• 对于另外 $20\%$ 的数据，$P_i=6$；
• 对于 $100\%$ 的数据，$1\le N\le 10^6$，$1\le P_i\le 6$。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll N=1e6+10;
ll p[N];

int main()
{
    ll n;
    cin>>n;

    ll nums=n*21;
    ll res=0;
    for(ll i=0;i<n;i++)
    {
        cin>>p[i];
        if(p[i]==1||p[i]==2)
            res+=3;
        if(p[i]==3||p[i]==4)
            res+=7;
        if(p[i]==5||p[i]==6)
            res+=11;
    }

    ll ans=nums-res+p[n-1];
    cout<<ans<<endl;
    return 0;
}