【做一道算一道】字节实习笔试（A出来的两道）

快乐氛围

题意大概是派对上每个人的氛围值不同，值高的跟值低的打招呼，总氛围会产生变化，值相同的打招呼不产生变化，求整体派对的氛围值。
氛围值变化公式=+高值-低值

示例

输入：
3
2 4 0
输出：
8
意为：
起始氛围为0；
4与2打招呼：总氛围+=4-2=2；
4与0打招呼：总氛围+=4-0=6；
2与0打招呼：总氛围+=2-0=8；

输入：
3
3 3 3
输出
0

思路

dp动态规划，先排序，相当于只有数值高的跟数值低的打招呼会增加氛围
排序后dp[0]=0,dp[1]=第二小的跟最小的差值
dp[i]的含义是与i之前的人打招呼增加的氛围值总和（和后面的不打）
从第三个开始算，相当于i当前的值和前一个的差值乘i相当于是概率里的Ci1（因为要跟之前的每个人打招呼，所以乘i）再加上前一个的dp就是当前的dp值。

代码

#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

int main() {
    long long n;
    cin>>n;
    vector<int> person(n);
    for(int i=0;i<n;i++){
        cin>>person[i];
    }
    sort(person.begin(), person.end());
    long long dp[n];

    if(n==1)
    cout<<0;
    else if(n==2)
    cout<<person[1]-person[0];
    else{
    dp[0]=0;
    dp[1]=person[1]-person[0];
    for(long long i=2;i<n;i++){
        dp[i]=(i)*(person[i]-person[i-1])+dp[i-1];
    }
    long long sum=0;
    for(auto i:dp){
        sum+=i;
    }
    cout<<sum;
    }
}
// 64 位输出请用 printf("%lld")

删除数字在第几轮

给一个数n和一个数k，生成从1到n的整数数组，每轮删去k的倍数位置的数字，直到数组元素数量小于k，求问第几轮删去n？

示例

输入：
13
2
输出：
3
意为：
1 2 3 4 5 6 7 8 9 10 11 12 13
第一轮删掉位置为2、4、6、8、10、12的数字
得到：1 3 5 7 9 11 13
第二轮删掉位置为2、4、6位置的数字
得到：1 5 9 13
第三轮删掉位置为2、4的数字
得到：1 9
第四轮删掉位置为2的数字
得到：1

输入：
4
5
输出：
0
删不了一点

思路

相当于只要能整除k的时候，n就被删掉了（说明是倍数），所以直接算
n/k是因为得到的就是每一轮次删除的数量，res=1是因为这里至少要进行一次删除。

代码

#include <iostream>
using namespace std;

int main() {
    long long n, k;
    cin >> n >> k;
    int res=1;

    if(n<k)
    cout<<0;
    else if(n%k==0)
    cout<<res;
    else{
    while (n%k) {
        n-=(n/k);
        res++;
    }
    cout<<res;
    }
}
// 64 位输出请用 printf("%lld")