首页 > 试题广场 >

编程题2

[编程题]编程题2
  • 热度指数:626 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 64M,其他语言128M
  • 算法知识视频讲解

有n个房间,现在i号房间里的人需要被重新分配,分配的规则是这样的:先让i号房间里的人全都出来,接下来按照 i+1, i+2, i+3, ... 的顺序依此往这些房间里放一个人,n号房间的的下一个房间是1号房间,直到所有的人都被重新分配。

现在告诉你分配完后每个房间的人数以及最后一个人被分配的房间号x,你需要求出分配前每个房间的人数。数据保证一定有解,若有多解输出任意一个解。


输入描述:
第一行两个整数n, x (2<=n<=10^5, 1<=x<=n),代表房间房间数量以及最后一个人被分配的房间号;
第二行n个整数 a_i(0<=a_i<=10^9) ,代表每个房间分配后的人数。


输出描述:
输出n个整数,代表每个房间分配前的人数。
示例1

输入

3 1
6 5 1

输出

4 4 4
const rl = require('readline').createInterface({
    input: process.stdin,
    output: process.stdout
})
let home = -1, tag = -1

let after = []

rl.on('line', input => {
    if (home == -1) {
        let temp = input.split(' ')
        home = Number.parseInt(temp[0])
        tag = Number.parseInt(temp[1])
    } else {
        after = input.split(' ').map(it => Number.parseInt(it))

        let min = after.reduce((x, y) => x <= y ? x : y, after[0])

        // min 代表循环经过 i 多少次
        // tag 是最后停在的位置

        for (let i = 0; i < after.length; i++) {
            after[i] -= min
        }

        // 此时的情况等同于分配轮数还没到一圈
        let start = tag - 1, cnt = 0
        while (after[start] != 0) {
            // ...
            cnt++
            after[start]--

            start--
            if (start == -1) {
                start = after.length - 1
            }
        }

        after[start] = home * min + cnt

        console.log(after.join(' '))
    }
    
})

发表于 2021-04-23 16:12:33 回复(0)

let [n, x] = readline().split(' ')
let arr = readline().split(' ')
let people = 0
let index = x - 1
let min = arr[0]
for (let i = 1; i < n; i ++) {
  if (Number(arr[i]) < Number(min)) {
    min = arr[i]
  }
}
arr = arr.map(item => item - min)
while(arr[index] !== 0 ) {
  arr[index] = arr[index] - 1
  people ++
  if (index === 0 ){
    index = n - 1
  } else {
    index --
  }
}
arr[index] = people + n*min
let res = ''
arr.forEach(item => {
   res += item + ' '
})
print(res)
编辑于 2019-04-14 09:14:01 回复(0)