// Změna na řešení s časovou složitostí O(n^2)
// Pro každou křižovatku kromě první určíme odpověď v konstantním čase
for (int r=0; r<N; ++r)
    for (int s=0; s<N; ++s) {
        if (r==0 && s==0) {
            while (soucet(r,s,ans[r][s]) < K)
	        ++ans[r][s];
        } else {
            int prevd = (s==0 ? ans[r-1][s] : ans[r][s-1]);
            for (int d = prevd-1; d <= prevd+1; ++d)
	        if (d>=0 && soucet(r,s,d) >= K)
		    { ans[r][s]=d; break; }
        }
    }