牛客寒假算法基础集训营3

牛客寒假算法基础集训营3

题目链接：https://ac.nowcoder.com/acm/contest/3004#question

A 牛牛的DRB迷宫I

题意：有一个n*m的迷宫，对于迷宫中的每个格子都为’R’,‘D’,'B’三种类型之一，'R’表示处于当前的格子时只能往右边走’D’表示处于当前的格子时只能往下边走，而’B’表示向右向下均可以走。

从左上角走到右下角不同种类的走法共有多少种，输出方案数对10^9+7109+7取余数后的结果。

思路：经典DP，从左上角开始循环，如果是D就往下累加，如果是R就往右累加，如果是B就同时累加。

$maze[i][j]{=}^{\prime }{D}^{\prime }dp[i+1][j]=dp[i+1][j]+dp[i][j]$

$maze[i][j]{=}^{\prime }{R}^{\prime }dp[i][j+1]=dp[i][j+1]+dp[i][j]$

$maze[i][j]{=}^{\prime }{B}^{\prime }dp[i+1][j]=dp[i+1][j]+dp[i][j],dp[i][j+1]=dp[i][j+1]+dp[i][j]$

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const long long mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;

ll dp[55][55];
int main()
{
    int n, m;
    char maze[55][55];
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= m; j++){
            cin >> maze[i][j];
        }
    }
    dp[1][1] = 1;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= m; j++){
            if(maze[i][j] == 'R'){
                dp[i][j + 1] = (dp[i][j] + dp[i][j + 1]) % mod;
            }
            else if(maze[i][j] == 'D'){
                dp[i + 1][j] = (dp[i][j] + dp[i + 1][j]) % mod;
            }
            else{
                dp[i][j + 1] = (dp[i][j] + dp[i][j + 1]) % mod;
                dp[i + 1][j] = (dp[i][j] + dp[i + 1][j]) % mod;
            }
        }
    }
    cout << dp[n][m] << endl;
    return 0;
}

C 牛牛的数组越位

数组越位，简单模拟

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const long long mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int e = 3e7 + 5;

ll s, n, m, p, x, y, val;
int a[e];

int main(){
    int t;
    cin >> t;
    while(t --){
        cin >> n >> m >> p;
        for(int i = 0; i < m * n; i++){
                a[i] = 0;
        }
        int k = 0;
        while(p --){
            cin >> x >> y >> val;
            ll g = x * m + y;
            if (x  < 0 || x >= n || y < 0 || y >= m){
                if (g < n * m && g >= 0){
                    a[g] = val;
                    if (k !=3) k = 1;
                }
                else k = 3;
            }
            else a[g] = val;
        }
        if(k == 3) cout << "Runtime error" << endl;
        else{
            int s = 0;
            for(int i = 0; i < m * n; i ++){
                s++;
                cout << a[i];
                if (s == m){
                    cout << endl;
                    s = 0;
                }
                else cout << " ";
            }
            if(!k) cout << "Accepted" << endl;
            else cout << "Undefined Behaviour" << endl;
        }
    }
    return 0;
}

D 牛牛与二叉树的数组存储

题意：数组模拟满二叉树，打印每个节点的父亲节点、左孩子、右孩子。对于下标为i的节点，他的左孩子的下标为i2，它的右孩子的下标为i2+1，它的父亲节点的下标为i/2。

需要按照value值从小到大的顺序进行输出。所以给value排序时，不能把节点序号弄乱。

多准备一个数组来存储节点序号。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const long long mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int e = 1e6 + 5;

typedef struct{int num, val;} test;
int n;
test tree1[e];
int tree[e];

bool mycmp(test t1, test t2){
    if(t1.val < t2.val) return 1;
    return 0;
}

int main()
{
    memset(tree, -1, sizeof tree);
    int siz = 0;
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> tree[i];
        if(tree[i] != -1){
            siz++;
            tree1[siz].val = tree[i];
            tree1[siz].num = i;
        }
    }
    sort(tree1 + 1, tree1 + siz + 1, mycmp);
    printf("The size of the tree is %d\n", siz);
    printf("Node %d is the root node of the tree\n", tree[1]);
    for(int i = 1; i <= siz; i++){
           // cout<<tree1[i].num<<" "<<tree1[i].val<<endl;
        printf("The father of node %d is %d, the left child is %d, and the right child is %d\n", i, tree[tree1[i].num / 2], tree[tree1[i].num * 2], tree[tree1[i].num * 2 + 1]);
    }
    return 0;
}

F 牛牛的Link Power I

题意：数组上的每一个节点都有两种状态，一种为link状态，另一种为cut状态。数组上任意一对处于link状态的无序点对（即(u,v)和(v,u)被认为是同一对）会产生dis(u,v)的link能量，dis(u,v)为数组上u到v的距离。

给出一个01串，'1’表示Link状态，'0’表示Cut状态，求整个数组的Link能量。

思路：两个1之间能量是对两个1的下标做减法，第一个1与第二个1之间的能量为第一个1的下标减去第二个1的下标；第三个1与第一个，第二个1之间的能量为，$3-1+3-2=2\ast 3-（1+2）$。以此类推，第n个1与其之前的1间的能量为 $（n-1）\ast n-(1+2+...+(n-1))$。

仅对1的下标做前缀和，并记录1的数量套公式即可

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const long long mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;

int main()
{
    int n;
    string s;
    cin >> n;
    cin >> s;
    ll num = 0, sum = 0, ans = 0;//num为1的数量  sum为1的前缀和
    for(int i = 1; i <= n; i++){
        if(s[i - 1] == '1'){
            ans = (ans + (((i * num) - sum) % mod)) % mod;
            num++;
            sum = (sum + i) % mod;
        }
    }
    cout << ans << endl;
    return 0;
}

H 牛牛的k合因子数

题意：“k合数”是指一个数的所有因子中，是合数的因子共有k个。求1~n中给定k的情况下k合因子数的数目。

思路：用筛法的思想，判断出质数合数，若为合数，找到将其作为因子的数，并将该数的合因子数目+1。判断完成后，若这次判断合因子数目为K，将K合因子数的数目+1，最后直接查询即可。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const long long mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int e = 1e5 + 5;

int n, m;
int prim[e],ans[e],temp[e];//prim为质数，ans中为K合因子数的数，temp中为合因子数的数目

void init(){
    for(int i = 2; i <= n; i++){
        prim[i] = 1;
    }
    for(int i = 2; i <= n; i++){
        if(prim[i]){
            //筛出质数和合数
            for(int j = i * 2; j <= n; j += i){
                prim[j] = 0;
            }
        }
        else{
            //若为合数，作为合因数向后寻找
            for(int j = i; j <= n; j += i){
                temp[j]++;
            }
        }
        ans[temp[i]]++;
    }
}

int main()
{
    cin >> n >> m;
    init();
    while(m--){
        int k;
        cin >> k;
        cout << ans[k] << endl;
    }
    return 0;
}

---