Submission #895551

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ld = long double;
using D = double;
using uint = unsigned int;

#ifdef WIN32
    #define LLD "%I64d"
#else
    #define LLD "%lld"
#endif

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second

const int maxn = 100005;

bool was[maxn];
int a[maxn];
int pos[maxn];
set<int> need[maxn];
set<int> waiting;
int ans[maxn], posans[maxn];
int n, m, q;

set<pair<int, int>> wass;

int main()
{
	scanf("%d%d", &n, &m);
	scanf("%d", &q);
	for (int i = 0; i < q; i++)
	{
		scanf("%d", &a[i]);
		a[i]--;
// 		scanf("%d", &a[m + i]);
// 		a[m + i]--;
	}
// 	q += m;
	for (int i = 0; i < m; i++) was[i] = false;
	for (int i = 0; i < n; i++)
	{
		need[a[q - 1]].insert(i);
		pos[i] = 0;
	}
	ans[0] = a[q - 1];
	was[a[q - 1]] = true;
	int anslen = 1;
	for (int i = q - 1; i >= 0; i--)
	{
		if (need[a[i]].size() > 0)
		{
			int wh = *need[a[i]].begin();
			need[a[i]].erase(need[a[i]].begin());
			pos[wh]++;
			if (anslen > pos[wh]) need[ans[pos[wh]]].insert(wh);
			else waiting.insert(wh);
			continue;
		}
		if (was[a[i]]) continue;
		ans[anslen] = a[i];
		anslen++;
		for (auto t : waiting)
		{
			need[a[i]].insert(t);
		}
		waiting.clear();
		was[a[i]] = true;
		{
			int wh = *need[a[i]].begin();
			need[a[i]].erase(need[a[i]].begin());
			pos[wh]++;
			if (anslen > pos[wh]) need[ans[pos[wh]]].insert(wh);
			else waiting.insert(wh);
		}
	}
// 	cout << anslen << endl;
// 	for (int i = 0; i < anslen; i++) cout << ans[i] << ' ';
// 	cout << endl;
// 	for (int i = 0; i < n; i++) cout << pos[i] << ' ';
// 	cout << endl;
	for (int i = 0; i < m; i++) posans[i] = m;
	for (int i = 0; i < anslen; i++) posans[ans[i]] = i;
	for (int i = 0; i < n; i++) if (pos[i] < anslen)
	{
// 		cout << i << ' ' << pos[i] << endl;
		int cur = 0;
		while (pos[i] < anslen && cur < m)
		{
// 			cout << cur << ' ' << posans[cur] << ' ' << pos[i] << endl;
			if (wass.count({pos[i], cur}))
			{
				pos[i] = anslen;
				break;
			}
			wass.insert({pos[i], cur});
			if (posans[cur] > pos[i])
			{
				printf("No\n");
				return 0;
			} else if (posans[cur] == pos[i])
			{
				pos[i]++;
				cur++;
			} else
			{
				cur++;
			}
		}
		if (pos[i] < anslen)
		{
			printf("No\n");
			return 0;
		}
	}
	printf("Yes\n");
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:36:23: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d%d", &n, &m);
                       ^
./Main.cpp:37:17: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &q);
                 ^
./Main.cpp:40:21: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d", &a[i]);
                     ^

Judge Result