#964302 (C++17) No.2698 Many Asakatsu

• 問題: No.2698 Many Asakatsu
• ユーザー: ゆにぽけ
• 提出日時: 2024-03-22 22:59:18
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,851 bytes
• コンパイル時間: 1,465 ms
• コンパイル使用メモリ: 132,660 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-09-30 12:21:08
• 合計ジャッジ時間: 5,183 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,816 KB
testcase_01	AC	2 ms; 6,816 KB
testcase_02	AC	2 ms; 6,816 KB
testcase_03	AC	2 ms; 6,816 KB
testcase_04	AC	2 ms; 6,820 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	AC	2 ms; 6,816 KB
testcase_08	AC	2 ms; 6,816 KB
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	AC	82 ms; 6,816 KB
testcase_30	AC	82 ms; 6,816 KB
testcase_31	WA	-
testcase_32	AC	226 ms; 6,820 KB
testcase_33	AC	44 ms; 6,816 KB
testcase_34	AC	220 ms; 6,816 KB

ソースコード

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#include <iterator>
#include <string>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <cassert>
#include <cmath>
#include <ctime>
#include <iomanip>
#include <numeric>
#include <stack>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <bitset>
#include <random>
#include <utility>
#include <functional>
using namespace std;
void Main()
{
	int N,M;
	cin >> N >> M;
	vector<int> A(N),B(N);
	for(int i = 0;i < N;i++)
	{
		cin >> A[i] >> B[i];
	}
	int Q;
	long long X;
	cin >> Q >> X;
	vector<int> C(N);
	for(int i = 0;i < N;i++)
	{
		cin >> C[i];
	}
	auto f = [&](long long x,int id) -> pair<long long,int>
	{
		long long ans = 0;
		int a = A[id],b = B[id];
		long long y = x - a;
		if(y < 0)
		{
			ans = (long long) M * (2 * a + (M - 1) * b) / 2 - x * M;
		}
		else if(y >= (M - 1) * b)
		{
			ans = -(long long) M * (2 * a + (M - 1) * b) / 2 + x * M;
		}
		else
		{
			//i * b <= x < (i + 1) * b
			//0 <= i < M - 1
			int j = y / b;
			ans += (long long) (j + 1) * (x - a) + (long long)(M - j - 1) * (a - x);
			ans += (long long) b * ((long long)M * (M - 1) / 2 - (long long)j * (j + 1) / 2);
			ans -= (long long) b * ((long long)j * (j + 1) / 2);
		}
		return make_pair(ans,id);
	};
	for(;Q--;)
	{
		int L = 0,R = N - 1;
		while(R - L > 2)
		{
			int mL = (L + L + R) / 3;
			int mR = (L + R + R) / 3;
			if(f(X,mL) < f(X,mR))
			{
				R = mR;
			}
			else
			{
				L = mL;
			}
		}
		pair<long long,int> a = f(X,L),b = f(X,L + 1),c = f(X,L + 2);
		if(a > b)
		{
			swap(a,b);
		}
		if(a > c)
		{
			swap(a,c);
		}
		cout << a.second + 1 << (Q ? " ":"\n");
		X += C[a.second];
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int tt = 1;
	/* cin >> tt; */
	while(tt--) Main();
}