https://codeforces.com/contest/670/problem/B
time limit per test
1 second
memory limit per test
256 megabytes
input
standard input
output
standard output
In late autumn evening n robots gathered in the cheerful company of friends. Each robot has a unique identifier — an integer from 1 to 109.
At some moment, robots decided to play the game "Snowball". Below there are the rules of this game. First, all robots stand in a row. Then the first robot says his identifier. After that the second robot says the identifier of the first robot and then says his own identifier. Then the third robot says the identifier of the first robot, then says the identifier of the second robot and after that says his own. This process continues from left to right until the n-th robot says his identifier.
Your task is to determine the k-th identifier to be pronounced.
Input
The first line contains two positive integers n and k (1 ≤ n ≤ 100 000, 1 ≤ k ≤ min(2·109, n·(n + 1) / 2).
The second line contains the sequence id1, id2, ..., idn (1 ≤ idi ≤ 109) — identifiers of roborts. It is guaranteed that all identifiers are different.
Output
Print the k-th pronounced identifier (assume that the numeration starts from 1).
Examples
input
2 2
1 2
output
1
input
4 5
10 4 18 3
output
4
Note
In the first sample identifiers of robots will be pronounced in the following order: 1, 1, 2. As k = 2, the answer equals to 1.
In the second test case identifiers of robots will be pronounced in the following order: 10, 10, 4, 10, 4, 18, 10, 4, 18, 3. As k = 5, the answer equals to 4.
题解:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <set>
#include <map>
using namespace std;
long long n,k,flag;
int a[100000+100];
int main()
{
scanf("%I64d%I64d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(long long i=1;i<=n;i++)
if(i*(i-1)/2+1<=k&&k<=i*(i+1)/2){
flag=i;
break;
}
cout << a[k-flag*(flag-1)/2] << endl;
//cout << "Hello world!" << endl;
return 0;
}