2023 (ICPC) Jiangxi Provincial Contest -- Official Contest J. Function

2023 (ICPC) Jiangxi Provincial Contest – Official Contest J. Function

题目链接：

J. Function

正经题意：

给出一个正整数 n(1 ⩽ n ⩽ 105) 和 n 个二次项系数为 1 的二次函数，第i 个函数形如 y = (x − i)^2 + bi(1 ⩽ bi ⩽ n)然后给出一个正整数 m(1 ⩽ m ⩽ 105)，表示操作总数。操作共有两种，第一种是添加一个二次项系数为 1 的二次函数，形如y = (x − a)^2 + b(1 ⩽ a, b ⩽ n)。第二种是询问所有二次函数在x = a(1 ⩽ a ⩽ n) 处的最小函数值。具体的，每次操作会先给出操作的类型，如果是 0 表示是第一种操作，如果是 1 表示是第二种操作。对于第一种操作，会再给出两个正整数 a, b。
对于第二种操作，则会再给出一个正整数 a。

简而言之：

给出了n个形如 y = (x − i)^2 + bi的函数，然后m个操作，第一种操作加一个**y = (x-a)^2+b **的函数，第二种操作找x=a时候最小函数值

思路：

容易发现，第一种操作来说，因为a是整数且在（1，n）范围内且查找的是最小函数值，所以不需要添加，只需要让 i=a时的b取最小，类似bi=min（b，bi）
第二种查找最小函数值，可以发现只需要查找距离a左右根号n的范围就可以了，因为如果距离大于了根号n，差值就大于了n，超过了b的最大值，不可能比等于a的时候更小

代码：

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

const int N = 2e6+27;

int main(){
	int n;
	scanf("%d",&n);
	for(int i = 1;i<=n;i++){
		scanf("%lld",&bb[i]);
	}
	int m;
	scanf("%d",&m);
	while(m--){
		int op;LL a,b;
		scanf("%d",&op);
		if(op==0){
			scanf("%lld%lld",&a,&b);
			bb[a] = min(bb[a],b);
		}else{
			scanf("%lld",&a);
			LL mina = 0x3f3f3f3f;
			for(LL i = max((LL)1,a-(LL)sqrt(n));i<=min((LL)n,a+(LL)sqrt(n));i++){
				mina = min(mina,(a-i)*(a-i)+bb[i]);
			}
			cout<<mina<<endl;
		}
	}
}